5c80fb77ffcdf8718257166e5a5ec7601d0b0cd9,deepchem/models/tensorgraph/tensor_graph.py,TensorGraph,fit_generator,#TensorGraph#Any#Any#Any#Any#Any#,149

Before Change

      if restore:
        self.restore()
      avg_loss, n_averaged_batches = 0.0, 0.0
      coord = tf.train.Coordinator()
      n_samples = 0
      n_enqueued = [0]
      final_sample = [None]
      if self.use_queue:

After Change

      if n_averaged_batches > 0:
        avg_loss = float(avg_loss) / n_averaged_batches
      if checkpoint_interval > 0:
        if n_averaged_batches > 0:
          print("Ending global_step %d: Average loss %g" % (self.global_step,
                                                            avg_loss))
        saver.save(self.session, self.save_file, global_step=self.global_step)
        time2 = time.time()
        print("TIMING: model fitting took %0.3f s" % (time2 - time1))
    return avg_loss

Instances