def train_model(
    *,
    model_config,
    debug = False,
    tb_log_dir = None,
    distribution_strategy = None,
    tpu = None):
  """Train an evidence matching model.

  Args:
    model_config: Set of parameters used for ModelConfig
    debug: Whether to enable debug features
    tb_log_dir: Where, if any, to log to tb
    distribution_strategy: CPU/GPU/TPU
    tpu: TPU config, if using TPU
  """
  trainer = training.Trainer(
      model_config,
      debug=debug,
      tpu=tpu,
      distribution_strategy=distribution_strategy,
      tb_log_dir=tb_log_dir)
  if debug:
    steps_per_epoch = 5
    validation_steps = 5
    epochs = 1
  else:
    steps_per_epoch = None
    validation_steps = None
    epochs = None
  trainer.train(
      epochs=epochs,
      steps_per_epoch=steps_per_epoch,
      validation_steps=validation_steps)
Train an evidence matching model.

  Args:
    model_config: Set of parameters used for ModelConfig
    debug: Whether to enable debug features
    tb_log_dir: Where, if any, to log to tb
    distribution_strategy: CPU/GPU/TPU
    tpu: TPU config, if using TPU
  
language/serene/fever_cli.py