diff --git a/video_prediction_tools/main_scripts/main_train_models.py b/video_prediction_tools/main_scripts/main_train_models.py index 47210f8cd69bd77d389232922361d4cb820aea5e..46874ab863aa5c2c5ac304b0df6c3986d8490fc5 100644 --- a/video_prediction_tools/main_scripts/main_train_models.py +++ b/video_prediction_tools/main_scripts/main_train_models.py @@ -264,7 +264,8 @@ class TrainModel(object): self.start_checkpoint_step = int(math.ceil(self.total_steps * self.frac_save_model_start)) self.saver_interval_step = int(math.ceil(self.total_steps * self.prob_save_model)) - + print("The model will be saved starting from step {} with {} interval step ".format(str(self.start_checkpoint_step),self.saver_interval_step)) + def restore(self, sess, checkpoints, restore_to_checkpoint_mapping=None): """