default save progress to true

lucidrains · Feb 27, 2021 · 1528492 · 1528492
1 parent df02f7d
commit 1528492
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/deep_daze/cli.py b/deep_daze/cli.py
@@ -17,7 +17,7 @@ def train(
         image_width=512,
         deeper=False,
         overwrite=False,
-        save_progress=False,
+        save_progress=True,
         seed=None,
         open_folder=True,
         save_date_time=False,

diff --git a/deep_daze/deep_daze.py b/deep_daze/deep_daze.py
@@ -246,7 +246,7 @@ def __init__(
             num_layers=16,
             epochs=20,
             iterations=1050,
-            save_progress=False,
+            save_progress=True,
             seed=None,
             open_folder=True,
             save_date_time=False,
@@ -414,8 +414,8 @@ def train_step(self, epoch, iteration):
                 out, loss = self.model(self.clip_encoding)
             loss = loss / self.gradient_accumulate_every
             total_loss += loss
-            self.scaler.scale(loss).backward()    
-        #out = normalize_image(out.cpu().float()).clamp(0., 1.)
+            self.scaler.scale(loss).backward()
+
         out = out.cpu().float().clamp(0., 1.)
         self.scaler.step(self.optimizer)
         self.scaler.update()

diff --git a/deep_daze/version.py b/deep_daze/version.py
@@ -1 +1 @@
-__version__ = '0.6.2'
+__version__ = '0.6.3'