def train(tfrecord_filename, estimator, num_train_steps):
  """Run one training loop over given TFRecords file."""
  train_input_fn = input_fn_builder(
      input_file=tfrecord_filename,
      seq_length=FLAGS.max_seq_length,
      qry_length=FLAGS.max_query_length,
      ent_length=FLAGS.max_entity_length,
      is_training=True,
      drop_remainder=True)
  estimator.train(input_fn=train_input_fn, max_steps=num_train_steps)
Run one training loop over given TFRecords file.
language/labs/drkit/run_dualencoder_lsf.py