From c38e7d49f2f559899923e325b2d7c787152b1c65 Mon Sep 17 00:00:00 2001 From: yerfor Date: Thu, 16 Mar 2023 14:41:24 +0800 Subject: [PATCH] update scripts --- scripts/train_audio2motion.sh | 2 +- scripts/train_postnet.sh | 2 +- scripts/train_radnerf.sh | 7 +++++++ 3 files changed, 9 insertions(+), 2 deletions(-) create mode 100644 scripts/train_radnerf.sh 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