remove blocking

autonomousvision · Jun 7, 2024 · 5b21b8d · 5b21b8d
1 parent 98d0858
commit 5b21b8d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/run_mipnerf360.py b/scripts/run_mipnerf360.py
@@ -18,7 +18,7 @@
 jobs = list(zip(scenes, factors))
 
 def train_scene(gpu, scene, factor):
-    cmd = f"CUDA_LAUNCH_BLOCKING=1 OMP_NUM_THREADS=4 CUDA_VISIBLE_DEVICES={gpu} python train.py -s 360_v2/{scene} -m {output_dir}/{scene} --eval -i images_{factor} --port {6109+int(gpu)}"
+    cmd = f"OMP_NUM_THREADS=4 CUDA_VISIBLE_DEVICES={gpu} python train.py -s 360_v2/{scene} -m {output_dir}/{scene} --eval -i images_{factor} --port {6109+int(gpu)}"
     print(cmd)
     if not dry_run:
        os.system(cmd)

diff --git a/scripts/run_tnt.py b/scripts/run_tnt.py
@@ -23,7 +23,7 @@
 jobs = list(zip(scenes, factors))
 
 def train_scene(gpu, scene, factor):
-    cmd = f"CUDA_LAUNCH_BLOCKING=1 OMP_NUM_THREADS=4 CUDA_VISIBLE_DEVICES={gpu} python train.py -s TNT_GOF/{split}/{scene} -m {output_dir}/{scene} --eval -r {factor} --use_decoupled_appearance"
+    cmd = f"OMP_NUM_THREADS=4 CUDA_VISIBLE_DEVICES={gpu} python train.py -s TNT_GOF/{split}/{scene} -m {output_dir}/{scene} --eval -r {factor} --use_decoupled_appearance"
     print(cmd)
     if not dry_run:
         os.system(cmd)