From 497a7eec94fea74898d86665a53f3c8606fc265b Mon Sep 17 00:00:00 2001 From: Colombo Date: Sun, 29 Mar 2020 14:50:02 +0400 Subject: [PATCH] fix preview_history --- models/ModelBase.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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():