diff --git a/models/ModelBase.py b/models/ModelBase.py index c23cd68..0ffe0c8 100644 --- 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():