migrate to cluster

chris-warner-II · Mar 22, 2024 · 9018e25 · 9018e25
1 parent c235596
commit 9018e25
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 7 deletions.
diff --git a/autoencoding.py b/autoencoding.py
@@ -16,19 +16,20 @@ def parse_args():
 
 def main():
 
-    args = parse_args()
-
+	args = parse_args()
+	print(args)
 
 	# # (1). Directory and device
 	dir_pre = 'store/models/diffae/'
 	dir_figs = 'store/output/diffae/autoencoding/'
 	os.makedirs(dir_figs,exist_ok=True)
 
 	device = 'mps' if torch.backends.mps.is_available() else 'cuda' if torch.cuda.is_available() else 'cpu'
-	device = 'cpu'
+	#device = 'cpu'
 	print(f'Using device: {device}')
 
-
+	if device=='cuda':
+		os.system('nvidia_smi')
 
 	# # (2). Setup and load in model
 	conf = ffhq256_autoenc()

diff --git a/imgs_manipulated/compare.png b/imgs_manipulated/compare.png
diff --git a/interpolate.py b/interpolate.py
@@ -25,9 +25,12 @@ def main():
     os.makedirs(dir_figs,exist_ok=True)
 
     device = 'mps' if torch.backends.mps.is_available() else 'cuda' if torch.cuda.is_available() else 'cpu'
-    device = 'cpu'
+    #device = 'cpu'
     print(f'Using device: {device}')
 
+    if device=='cuda':
+        os.system('nvidia_smi')
+
 
     # # (2). Setup and load in model
     conf = ffhq256_autoenc()

diff --git a/manipulate.py b/manipulate.py
@@ -27,9 +27,12 @@ def main():
     os.makedirs(dir_figs,exist_ok=True)
 
     device = 'mps' if torch.backends.mps.is_available() else 'cuda' if torch.cuda.is_available() else 'cpu'
-    device = 'cpu'
+    #device = 'cpu'
     print(f'Using device: {device}')
 
+    if device=='cuda':
+        os.system('nvidia_smi')
+
 
     # # (2). Setup and load in models
     conf = ffhq256_autoenc()

diff --git a/sample.py b/sample.py
@@ -22,9 +22,12 @@ def main():
     os.makedirs(dir_figs,exist_ok=True)
 
     device = 'mps' if torch.backends.mps.is_available() else 'cuda' if torch.cuda.is_available() else 'cpu'
-    device = 'cpu'
+    #device = 'cpu'
     print(f'Using device: {device}')
 
+    if device=='cuda':
+        os.system('nvidia_smi')
+
 
     # # (2). Setup and load in model
     conf = ffhq256_autoenc_latent()