read seedValue automatically

main_sphere_compression.py

@@ -432,11 +432,17 @@ def main():
         print(f"{key:_<40}: {val}\n")  # check this for all kinds of formatting
     print("=" * 60 + "\n")
 
+    seedValue = random.randrange(2**32)   # create a seed
     if args.seed is not None:
         seedValue = args.seed
-    else:
-        seedValue = random.randrange(2**32)   # create a seed
+    elif args.checkpoint_file and os.path.isfile(os.path.join(args.out_dir, 'config.txt')):
+        with open(os.path.join(args.out_dir, 'config.txt'), 'r') as f:
+            config_lines = f.readlines()
+            for line in config_lines:
+                if line.strip().startswith('seedValue'):
+                    seedValue = int(line.split(': ')[1])
     common_pytorch.set_seed(seedValue)
+    
     device_str = 'cpu'
     if args.gpu and torch.cuda.is_available():
         device_str = 'cuda'