diff --git a/video_prediction_tools/main_scripts/main_train_models.py b/video_prediction_tools/main_scripts/main_train_models.py index 6ee98cfade84c15575117010945d57d383adac19..98df062bf9919efa0657c8a63a8a842a54d2502b 100644 --- a/video_prediction_tools/main_scripts/main_train_models.py +++ b/video_prediction_tools/main_scripts/main_train_models.py @@ -318,6 +318,7 @@ class TrainModel(object): self.create_fetches_for_val() self.val_results = sess.run(self.val_fetches, feed_dict={self.train_handle: val_handle_eval}) val_losses.append(self.val_results[list(self.saver_loss)[0]]) + print(val_losses) self.write_to_summary() self.print_results(step, self.results) # track iteration time