update scripts

yerfor · Mar 16, 2023 · c38e7d4 · c38e7d4
1 parent 90bffaa
commit c38e7d4
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/scripts/train_audio2motion.sh b/scripts/train_audio2motion.sh
@@ -1 +1 @@
-CUDA_VISIBLE_DEVICES=0 python tasks/run.py --config=egs/datasets/lrs3/lm3d_vae_sync.yaml --exp_name=lrs3/lm3d_vae --reset
+CUDA_VISIBLE_DEVICES=0 python tasks/run.py --config=egs/datasets/lrs3/lm3d_vae_sync.yaml --exp_name=lrs3/lm3d_vae_sync --reset
diff --git a/scripts/train_postnet.sh b/scripts/train_postnet.sh
@@ -1,2 +1,2 @@
 export Video_ID=May
-CUDA_VISIBLE_DEVICES=0 python tasks/run.py --config=egs/datasets/videos/${Video_ID}/lm3d_postnet_sync.yaml --exp_name=${Video_ID}/postnet --reset
+CUDA_VISIBLE_DEVICES=0 python tasks/run.py --config=egs/datasets/videos/${Video_ID}/lm3d_postnet_sync.yaml --exp_name=${Video_ID}/lm3d_postnet_sync --reset
diff --git a/scripts/train_radnerf.sh b/scripts/train_radnerf.sh
@@ -0,0 +1,7 @@
+export Video_ID=May
+# binarize the dataset
+python data_gen/nerf/binarizer.py --config=egs/datasets/videos/${Video_ID}/lm3d_radnerf.yaml
+# train Head NeRF
+CUDA_VISIBLE_DEVICES=0 python tasks/run.py --config=egs/datasets/videos/${Video_ID}/lm3d_radnerf.yaml --exp_name=${Video_ID}/lm3d_radnerf --reset
+# train Torso NeRF
+CUDA_VISIBLE_DEVICES=0 python tasks/run.py --config=egs/datasets/videos/${Video_ID}/lm3d_radnerf_torso.yaml --exp_name=${Video_ID}/lm3d_radnerf_torso --reset