remove image tensor after initial priming training

lucidrains · Feb 1, 2021 · 2f873f8 · 2f873f8
1 parent 3bcdfdd
commit 2f873f8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/deep_daze/deep_daze.py b/deep_daze/deep_daze.py
@@ -333,6 +333,9 @@ def forward(self):
                     print('interrupted by keyboard, gracefully exiting')
                     return exit()
 
+            del self.start_image
+            del optim
+
         tqdm.write(f'Imagining "{self.text}" from the depths of my weights...')
 
         if self.open_folder:

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