fix preview_history

2020-03-29 14:50:02 +04:00
parent 4e744cf184
commit 497a7eec94
1 changed files with 2 additions and 1 deletions
--- a/models/ModelBase.py
+++ b/models/ModelBase.py
@@ -434,7 +434,8 @@ class ModelBase(object):

        self.loss_history.append ( [float(loss[1]) for loss in losses] )

-        if self.iter % 100 == 0:
+        if (not io.is_colab() and self.iter % 10 == 0) or \
+           (io.is_colab() and self.iter % 100 == 0):           
            plist = []

            if io.is_colab():