diff --git a/scripts/train_audio2motion.sh b/scripts/train_audio2motion.sh index 3cb3dc5b..c32a61a8 100644 --- 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 index c39f627f..073474ee 100644 --- 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 new file mode 100644 index 00000000..ec70d569 --- /dev/null +++ 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 \ No newline at end of file