def train(self, dset):
    """Train the model on this dataset over the examples in a dataset.

    Args:
      dset: a tf.data.Dataset
    """
    start_time = time.time()
    for fd in self.feed_dict_iterator(dset):
      _, latest_loss = self.session.run([self.model.train_op, self.model.loss],
                                        feed_dict=fd)
      if self.model.training_callback is not None:
        status = self.model.training_callback(fd, latest_loss,
                                              time.time() - start_time)
        if status:
          logging.info(status)
Train the model on this dataset over the examples in a dataset.

    Args:
      dset: a tf.data.Dataset
    
language/nql/nql/util.py