mirror what is done in big sleep

Author: lucidrains

deep_daze/cli.py

@@ -19,6 +19,8 @@ def train(
     seed = None,
     open_folder = True
 ):
+    print('Starting up...')
+
     if deeper:
         num_layers = 32
 

deep_daze/deep_daze.py

@@ -208,6 +208,9 @@ def __init__(
         if exists(seed):
             print(f'setting seed: {seed}')
             torch.manual_seed(seed)
+            torch.cuda.manual_seed(seed)
+            random.seed(seed)
+            torch.backends.cudnn.deterministic=True
 
         self.epochs = epochs
         self.iterations = iterations
@@ -270,6 +273,7 @@ def forward(self):
 
         if self.open_folder:
             open_folder('./')
+            self.open_folder = False
 
         for epoch in trange(self.epochs, desc = 'epochs'):
             pbar = trange(self.iterations, desc='iteration')

deep_daze/version.py

@@ -1 +1 @@
-__version__ = '0.2.9'
+__version__ = '0.3.0'