From 5d6d7d35dba09bea6135119a1398bf1af8c96cf8 Mon Sep 17 00:00:00 2001
From: gong1 <b.gong@fz-juelich.de>
Date: Mon, 7 Feb 2022 19:27:47 +0100
Subject: [PATCH] fix bugs
---
video_prediction_tools/main_scripts/main_train_models.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/video_prediction_tools/main_scripts/main_train_models.py b/video_prediction_tools/main_scripts/main_train_models.py
index 707085b6..9e58de96 100644
--- a/video_prediction_tools/main_scripts/main_train_models.py
+++ b/video_prediction_tools/main_scripts/main_train_models.py
@@ -247,6 +247,10 @@ class TrainModel(object):
self.steps_per_epoch = int(self.num_examples/self.batch_size)
self.total_steps = self.steps_per_epoch * self.max_epochs
self.diag_intv_step = int(self.diag_intv_frac*self.total_steps)
+ if self.diag_intv_step == 0:
+ self.diag_intv_step = 1
+ else:
+ pass
print("%{}: Batch size: {}; max_epochs: {}; num_samples per epoch: {}; steps_per_epoch: {}, total steps: {}"
.format(method, self.batch_size, self.max_epochs, self.num_examples, self.steps_per_epoch,
self.total_steps))
--
GitLab