diff --git a/scripts/run_mipnerf360.py b/scripts/run_mipnerf360.py index f7bba43..2721e0c 100644 --- 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 index b9f81a4..5ce51bb 100644 --- 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)