Upload 28 files
Browse files- sample_scripts/sampling_cond_broken.sh +18 -0
- sample_scripts/sampling_cond_multiplane.sh +23 -0
- sample_scripts/sampling_cond_multiplane_broken.sh +17 -0
- sample_scripts/sampling_cond_oneplane.sh +24 -0
- sample_scripts/sampling_cond_random.sh +17 -0
- sample_scripts/sampling_cond_random_eval_only.sh +17 -0
- sample_scripts/sampling_cond_triplane.sh +25 -0
- sample_scripts/sampling_cond_triplane_broken.sh +23 -0
- sample_scripts/sampling_cond_triplane_multiplane.sh +25 -0
- sample_scripts/sampling_cond_triplane_multiplane_broken.sh +23 -0
- sample_scripts/sampling_guide_broken.sh +18 -0
- sample_scripts/sampling_guide_multiplane.sh +23 -0
- sample_scripts/sampling_guide_multiplane_broken.sh +23 -0
- sample_scripts/sampling_guide_oneplane.sh +17 -0
- sample_scripts/sampling_guide_random.sh +17 -0
- sample_scripts/sampling_guide_triplane.sh +23 -0
- sample_scripts/sampling_guide_triplane_broken.sh +23 -0
- sample_scripts/sampling_guide_triplane_multiplane.sh +23 -0
- sample_scripts/sampling_guide_triplane_multiplane_broken.sh +23 -0
- sample_scripts/sampling_uncond.sh +19 -0
- sample_scripts/zero_shot_sampling_cond_broken.sh +18 -0
- sample_scripts/zero_shot_sampling_cond_multiplane.sh +17 -0
- sample_scripts/zero_shot_sampling_cond_oneplane.sh +17 -0
- sample_scripts/zero_shot_sampling_cond_triplane.sh +17 -0
- sample_scripts/zero_shot_sampling_guide_broken.sh +18 -0
- sample_scripts/zero_shot_sampling_guide_multiplane.sh +17 -0
- sample_scripts/zero_shot_sampling_guide_oneplane.sh +17 -0
- sample_scripts/zero_shot_sampling_guide_triplane.sh +17 -0
sample_scripts/sampling_cond_broken.sh
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 3 |
+
dataset.name='msn' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=False \
|
| 9 |
+
model.use_mask_cond=True \
|
| 10 |
+
model.cond_num=1 \
|
| 11 |
+
model.use_oneplane=False \
|
| 12 |
+
model.use_triplane=False \
|
| 13 |
+
model.use_multiplane=False \
|
| 14 |
+
model.use_broken=True \
|
| 15 |
+
model.output_dir='output/DDIM_V2__COND1__use_broken__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND1__use_broken/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
| 18 |
+
|
sample_scripts/sampling_cond_multiplane.sh
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
export http_proxy=http://127.0.0.1:1082
|
| 2 |
+
export https_proxy=http://127.0.0.1:1082
|
| 3 |
+
export HTTP_PROXY=http://127.0.0.1:1082
|
| 4 |
+
export HTTPS_PROXY=http://127.0.0.1:1082
|
| 5 |
+
unset all_proxy ALL_PROXY
|
| 6 |
+
|
| 7 |
+
CUDA_VISIBLE_DEVICES=0 python sample_msn_v3.py \
|
| 8 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 9 |
+
dataset.name='msn' \
|
| 10 |
+
dataset.batch_size=12 \
|
| 11 |
+
dataset.minibatch_size=4 \
|
| 12 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 13 |
+
model.timesteps=50 \
|
| 14 |
+
model.use_mask_guide=False \
|
| 15 |
+
model.use_mask_cond=True \
|
| 16 |
+
model.cond_num=1 \
|
| 17 |
+
model.use_oneplane=False \
|
| 18 |
+
model.use_triplane=False \
|
| 19 |
+
model.use_multiplane=True \
|
| 20 |
+
model.use_broken=False \
|
| 21 |
+
model.output_dir='output-farm02/COND1__use_multiplane__cp20' \
|
| 22 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND1__use_multiplane/model-20.pt' \
|
| 23 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_cond_multiplane_broken.sh
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=0 python sample_msn_v3.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 3 |
+
dataset.name='msn' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=False \
|
| 9 |
+
model.use_mask_cond=True \
|
| 10 |
+
model.cond_num=2 \
|
| 11 |
+
model.use_oneplane=False \
|
| 12 |
+
model.use_triplane=False \
|
| 13 |
+
model.use_multiplane=True \
|
| 14 |
+
model.use_broken=True \
|
| 15 |
+
model.output_dir='output/COND2__use_multiplane__use_broken__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND2__use_multiplane__use_broken/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_cond_oneplane.sh
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
export http_proxy=http://127.0.0.1:1082
|
| 2 |
+
export https_proxy=http://127.0.0.1:1082
|
| 3 |
+
export HTTP_PROXY=http://127.0.0.1:1082
|
| 4 |
+
export HTTPS_PROXY=http://127.0.0.1:1082
|
| 5 |
+
unset all_proxy ALL_PROXY
|
| 6 |
+
|
| 7 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3.py \
|
| 8 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 9 |
+
dataset.name='msn' \
|
| 10 |
+
dataset.batch_size=12 \
|
| 11 |
+
dataset.minibatch_size=4 \
|
| 12 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 13 |
+
model.timesteps=50 \
|
| 14 |
+
model.use_mask_guide=False \
|
| 15 |
+
model.use_mask_cond=True \
|
| 16 |
+
model.cond_num=1 \
|
| 17 |
+
model.use_oneplane=True \
|
| 18 |
+
model.use_triplane=False \
|
| 19 |
+
model.use_multiplane=False \
|
| 20 |
+
model.use_broken=False \
|
| 21 |
+
model.output_dir='output-farm02/DDIM_V3__COND1__use_oneplane__cp20' \
|
| 22 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND1__use_oneplane/model-20.pt' \
|
| 23 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
| 24 |
+
|
sample_scripts/sampling_cond_random.sh
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3_random.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 3 |
+
dataset.name='msn' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=False \
|
| 9 |
+
model.use_mask_cond=True \
|
| 10 |
+
model.cond_num=3 \
|
| 11 |
+
model.use_oneplane=False \
|
| 12 |
+
model.use_triplane=True \
|
| 13 |
+
model.use_multiplane=True \
|
| 14 |
+
model.use_broken=True \
|
| 15 |
+
model.output_dir='output/COND_RANDOM__COND3__use_triplane__use_multiplane__use_broken__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/COND_RANDOM/LR0.001__BS24__TS50__COND3__use_triplane__use_multiplane__use_broken/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_cond_random_eval_only.sh
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3_random_eval_only.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 3 |
+
dataset.name='msn' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=False \
|
| 9 |
+
model.use_mask_cond=True \
|
| 10 |
+
model.cond_num=3 \
|
| 11 |
+
model.use_oneplane=False \
|
| 12 |
+
model.use_triplane=True \
|
| 13 |
+
model.use_multiplane=True \
|
| 14 |
+
model.use_broken=True \
|
| 15 |
+
model.output_dir='output/COND_RANDOM_EVAL_ONLY__COND3__use_triplane__use_multiplane__use_broken__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND3__use_triplane__use_multiplane__use_broken/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_cond_triplane.sh
ADDED
|
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
sleep 14400
|
| 2 |
+
|
| 3 |
+
export http_proxy=http://127.0.0.1:1082
|
| 4 |
+
export https_proxy=http://127.0.0.1:1082
|
| 5 |
+
export HTTP_PROXY=http://127.0.0.1:1082
|
| 6 |
+
export HTTPS_PROXY=http://127.0.0.1:1082
|
| 7 |
+
unset all_proxy ALL_PROXY
|
| 8 |
+
|
| 9 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3.py \
|
| 10 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 11 |
+
dataset.name='msn' \
|
| 12 |
+
dataset.batch_size=12 \
|
| 13 |
+
dataset.minibatch_size=4 \
|
| 14 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 15 |
+
model.timesteps=50 \
|
| 16 |
+
model.use_mask_guide=False \
|
| 17 |
+
model.use_mask_cond=True \
|
| 18 |
+
model.cond_num=1 \
|
| 19 |
+
model.use_oneplane=False \
|
| 20 |
+
model.use_triplane=True \
|
| 21 |
+
model.use_multiplane=False \
|
| 22 |
+
model.use_broken=False \
|
| 23 |
+
model.output_dir='output-farm02/COND1__use_triplane__cp20' \
|
| 24 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND1__use_triplane/model-20.pt' \
|
| 25 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_cond_triplane_broken.sh
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
export http_proxy=http://127.0.0.1:1082
|
| 2 |
+
export https_proxy=http://127.0.0.1:1082
|
| 3 |
+
export HTTP_PROXY=http://127.0.0.1:1082
|
| 4 |
+
export HTTPS_PROXY=http://127.0.0.1:1082
|
| 5 |
+
unset all_proxy ALL_PROXY
|
| 6 |
+
|
| 7 |
+
CUDA_VISIBLE_DEVICES=0 python sample_msn_v3.py \
|
| 8 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 9 |
+
dataset.name='msn' \
|
| 10 |
+
dataset.batch_size=12 \
|
| 11 |
+
dataset.minibatch_size=4 \
|
| 12 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 13 |
+
model.timesteps=50 \
|
| 14 |
+
model.use_mask_guide=False \
|
| 15 |
+
model.use_mask_cond=True \
|
| 16 |
+
model.cond_num=2 \
|
| 17 |
+
model.use_oneplane=False \
|
| 18 |
+
model.use_triplane=True \
|
| 19 |
+
model.use_multiplane=False \
|
| 20 |
+
model.use_broken=True \
|
| 21 |
+
model.output_dir='output-farm02/COND2__use_triplane__use_broken__cp20' \
|
| 22 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND2__use_triplane__use_broken/model-20.pt' \
|
| 23 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_cond_triplane_multiplane.sh
ADDED
|
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
sleep 14400
|
| 2 |
+
|
| 3 |
+
export http_proxy=http://127.0.0.1:1082
|
| 4 |
+
export https_proxy=http://127.0.0.1:1082
|
| 5 |
+
export HTTP_PROXY=http://127.0.0.1:1082
|
| 6 |
+
export HTTPS_PROXY=http://127.0.0.1:1082
|
| 7 |
+
unset all_proxy ALL_PROXY
|
| 8 |
+
|
| 9 |
+
CUDA_VISIBLE_DEVICES=0 python sample_msn_v3.py \
|
| 10 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 11 |
+
dataset.name='msn' \
|
| 12 |
+
dataset.batch_size=12 \
|
| 13 |
+
dataset.minibatch_size=4 \
|
| 14 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 15 |
+
model.timesteps=50 \
|
| 16 |
+
model.use_mask_guide=False \
|
| 17 |
+
model.use_mask_cond=True \
|
| 18 |
+
model.cond_num=2 \
|
| 19 |
+
model.use_oneplane=False \
|
| 20 |
+
model.use_triplane=True \
|
| 21 |
+
model.use_multiplane=True \
|
| 22 |
+
model.use_broken=False \
|
| 23 |
+
model.output_dir='output-farm02/COND2__use_triplane__use_multiplane__cp20' \
|
| 24 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND2__use_triplane__use_multiplane/model-20.pt' \
|
| 25 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_cond_triplane_multiplane_broken.sh
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
export http_proxy=http://127.0.0.1:1082
|
| 2 |
+
export https_proxy=http://127.0.0.1:1082
|
| 3 |
+
export HTTP_PROXY=http://127.0.0.1:1082
|
| 4 |
+
export HTTPS_PROXY=http://127.0.0.1:1082
|
| 5 |
+
unset all_proxy ALL_PROXY
|
| 6 |
+
|
| 7 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3.py \
|
| 8 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 9 |
+
dataset.name='msn' \
|
| 10 |
+
dataset.batch_size=12 \
|
| 11 |
+
dataset.minibatch_size=4 \
|
| 12 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 13 |
+
model.timesteps=50 \
|
| 14 |
+
model.use_mask_guide=False \
|
| 15 |
+
model.use_mask_cond=True \
|
| 16 |
+
model.cond_num=3 \
|
| 17 |
+
model.use_oneplane=False \
|
| 18 |
+
model.use_triplane=True \
|
| 19 |
+
model.use_multiplane=True \
|
| 20 |
+
model.use_broken=True \
|
| 21 |
+
model.output_dir='output-farm02/COND3__use_triplane__use_multiplane__use_broken__cp20' \
|
| 22 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND3__use_triplane__use_multiplane__use_broken/model-20.pt' \
|
| 23 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_guide_broken.sh
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=0 python sample_msn_v3.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 3 |
+
dataset.name='msn' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=True \
|
| 9 |
+
model.use_mask_cond=False \
|
| 10 |
+
model.cond_num=0 \
|
| 11 |
+
model.use_oneplane=False \
|
| 12 |
+
model.use_triplane=False \
|
| 13 |
+
model.use_multiplane=False \
|
| 14 |
+
model.use_broken=True \
|
| 15 |
+
model.output_dir='output/DDIM_V2__GUIDE1__use_broken__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
| 18 |
+
|
sample_scripts/sampling_guide_multiplane.sh
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
export http_proxy=http://127.0.0.1:1081
|
| 2 |
+
export https_proxy=http://127.0.0.1:1081
|
| 3 |
+
export HTTP_PROXY=http://127.0.0.1:1081
|
| 4 |
+
export HTTPS_PROXY=http://127.0.0.1:1081
|
| 5 |
+
unset all_proxy ALL_PROXY
|
| 6 |
+
|
| 7 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3.py \
|
| 8 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 9 |
+
dataset.name='msn' \
|
| 10 |
+
dataset.batch_size=12 \
|
| 11 |
+
dataset.minibatch_size=8 \
|
| 12 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 13 |
+
model.timesteps=50 \
|
| 14 |
+
model.use_mask_guide=True \
|
| 15 |
+
model.use_mask_cond=False \
|
| 16 |
+
model.cond_num=0 \
|
| 17 |
+
model.use_oneplane=False \
|
| 18 |
+
model.use_triplane=False \
|
| 19 |
+
model.use_multiplane=True \
|
| 20 |
+
model.use_broken=False \
|
| 21 |
+
model.output_dir='output-farm02/GUIDE1__use_multiplane__cp20' \
|
| 22 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 23 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_guide_multiplane_broken.sh
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# export http_proxy=http://127.0.0.1:1081
|
| 2 |
+
# export https_proxy=http://127.0.0.1:1081
|
| 3 |
+
# export HTTP_PROXY=http://127.0.0.1:1081
|
| 4 |
+
# export HTTPS_PROXY=http://127.0.0.1:1081
|
| 5 |
+
# unset all_proxy ALL_PROXY
|
| 6 |
+
|
| 7 |
+
CUDA_VISIBLE_DEVICES=0 python sample_msn_v3.py \
|
| 8 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 9 |
+
dataset.name='msn' \
|
| 10 |
+
dataset.batch_size=12 \
|
| 11 |
+
dataset.minibatch_size=6 \
|
| 12 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 13 |
+
model.timesteps=50 \
|
| 14 |
+
model.use_mask_guide=True \
|
| 15 |
+
model.use_mask_cond=False \
|
| 16 |
+
model.cond_num=0 \
|
| 17 |
+
model.use_oneplane=False \
|
| 18 |
+
model.use_triplane=False \
|
| 19 |
+
model.use_multiplane=True \
|
| 20 |
+
model.use_broken=True \
|
| 21 |
+
model.output_dir='output/GUIDE2__use_multiplane__use_broken__cp20' \
|
| 22 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 23 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_guide_oneplane.sh
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 3 |
+
dataset.name='msn' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=6 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=True \
|
| 9 |
+
model.use_mask_cond=False \
|
| 10 |
+
model.cond_num=0 \
|
| 11 |
+
model.use_oneplane=True \
|
| 12 |
+
model.use_triplane=False \
|
| 13 |
+
model.use_multiplane=False \
|
| 14 |
+
model.use_broken=False \
|
| 15 |
+
model.output_dir='output/GUIDE1__use_oneplane__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_guide_random.sh
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=0 python sample_msn_v3_random.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 3 |
+
dataset.name='msn' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=6 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=True \
|
| 9 |
+
model.use_mask_cond=False \
|
| 10 |
+
model.cond_num=0 \
|
| 11 |
+
model.use_oneplane=False \
|
| 12 |
+
model.use_triplane=True \
|
| 13 |
+
model.use_multiplane=True \
|
| 14 |
+
model.use_broken=True \
|
| 15 |
+
model.output_dir='output/GUIDE_RANDOM__GUIDE3__use_triplane__use_multiplane__use_broken__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_guide_triplane.sh
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# export http_proxy=http://127.0.0.1:1081
|
| 2 |
+
# export https_proxy=http://127.0.0.1:1081
|
| 3 |
+
# export HTTP_PROXY=http://127.0.0.1:1081
|
| 4 |
+
# export HTTPS_PROXY=http://127.0.0.1:1081
|
| 5 |
+
# unset all_proxy ALL_PROXY
|
| 6 |
+
|
| 7 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3.py \
|
| 8 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 9 |
+
dataset.name='msn' \
|
| 10 |
+
dataset.batch_size=12 \
|
| 11 |
+
dataset.minibatch_size=6 \
|
| 12 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 13 |
+
model.timesteps=50 \
|
| 14 |
+
model.use_mask_guide=True \
|
| 15 |
+
model.use_mask_cond=False \
|
| 16 |
+
model.cond_num=0 \
|
| 17 |
+
model.use_oneplane=False \
|
| 18 |
+
model.use_triplane=True \
|
| 19 |
+
model.use_multiplane=False \
|
| 20 |
+
model.use_broken=False \
|
| 21 |
+
model.output_dir='output/GUIDE1__use_triplane__cp20' \
|
| 22 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 23 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_guide_triplane_broken.sh
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
export http_proxy=http://127.0.0.1:1082
|
| 2 |
+
export https_proxy=http://127.0.0.1:1082
|
| 3 |
+
export HTTP_PROXY=http://127.0.0.1:1082
|
| 4 |
+
export HTTPS_PROXY=http://127.0.0.1:1082
|
| 5 |
+
unset all_proxy ALL_PROXY
|
| 6 |
+
|
| 7 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3.py \
|
| 8 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 9 |
+
dataset.name='msn' \
|
| 10 |
+
dataset.batch_size=12 \
|
| 11 |
+
dataset.minibatch_size=6 \
|
| 12 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 13 |
+
model.timesteps=50 \
|
| 14 |
+
model.use_mask_guide=True \
|
| 15 |
+
model.use_mask_cond=False \
|
| 16 |
+
model.cond_num=0 \
|
| 17 |
+
model.use_oneplane=False \
|
| 18 |
+
model.use_triplane=True \
|
| 19 |
+
model.use_multiplane=False \
|
| 20 |
+
model.use_broken=True \
|
| 21 |
+
model.output_dir='output-farm02/GUIDE2__use_triplane__use_broken__cp20' \
|
| 22 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 23 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_guide_triplane_multiplane.sh
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
export http_proxy=http://127.0.0.1:1082
|
| 2 |
+
export https_proxy=http://127.0.0.1:1082
|
| 3 |
+
export HTTP_PROXY=http://127.0.0.1:1082
|
| 4 |
+
export HTTPS_PROXY=http://127.0.0.1:1082
|
| 5 |
+
unset all_proxy ALL_PROXY
|
| 6 |
+
|
| 7 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3.py \
|
| 8 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 9 |
+
dataset.name='msn' \
|
| 10 |
+
dataset.batch_size=12 \
|
| 11 |
+
dataset.minibatch_size=6 \
|
| 12 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 13 |
+
model.timesteps=50 \
|
| 14 |
+
model.use_mask_guide=True \
|
| 15 |
+
model.use_mask_cond=False \
|
| 16 |
+
model.cond_num=0 \
|
| 17 |
+
model.use_oneplane=False \
|
| 18 |
+
model.use_triplane=True \
|
| 19 |
+
model.use_multiplane=True \
|
| 20 |
+
model.use_broken=False \
|
| 21 |
+
model.output_dir='output-farm02/GUIDE3__use_triplane__use_multiplane__cp20' \
|
| 22 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 23 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_guide_triplane_multiplane_broken.sh
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
export http_proxy=http://127.0.0.1:1082
|
| 2 |
+
export https_proxy=http://127.0.0.1:1082
|
| 3 |
+
export HTTP_PROXY=http://127.0.0.1:1082
|
| 4 |
+
export HTTPS_PROXY=http://127.0.0.1:1082
|
| 5 |
+
unset all_proxy ALL_PROXY
|
| 6 |
+
|
| 7 |
+
CUDA_VISIBLE_DEVICES=0 python sample_msn_v3.py \
|
| 8 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 9 |
+
dataset.name='msn' \
|
| 10 |
+
dataset.batch_size=12 \
|
| 11 |
+
dataset.minibatch_size=6 \
|
| 12 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 13 |
+
model.timesteps=50 \
|
| 14 |
+
model.use_mask_guide=True \
|
| 15 |
+
model.use_mask_cond=False \
|
| 16 |
+
model.cond_num=0 \
|
| 17 |
+
model.use_oneplane=False \
|
| 18 |
+
model.use_triplane=True \
|
| 19 |
+
model.use_multiplane=True \
|
| 20 |
+
model.use_broken=True \
|
| 21 |
+
model.output_dir='output-farm02/GUIDE3__use_triplane__use_multiplane__use_broken__cp20' \
|
| 22 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 23 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/sampling_uncond.sh
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
|
| 3 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_uncond.py \
|
| 4 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/dataset/medshapenet/data_in_category_with_one_component_crop_zoom56_pad64_sdf_new_64' \
|
| 5 |
+
dataset.name='msn' \
|
| 6 |
+
dataset.batch_size=16 \
|
| 7 |
+
dataset.minibatch_size=4 \
|
| 8 |
+
dataset.text_embedding_file='data_preproc/claude_sonnet_4.5_prompt_qwen3_32b_embedding.json' \
|
| 9 |
+
model.timesteps=50 \
|
| 10 |
+
model.use_mask_guide=False \
|
| 11 |
+
model.use_mask_cond=False \
|
| 12 |
+
model.cond_num=0 \
|
| 13 |
+
model.use_oneplane=False \
|
| 14 |
+
model.use_triplane=False \
|
| 15 |
+
model.use_multiplane=False \
|
| 16 |
+
model.use_broken=False \
|
| 17 |
+
model.output_dir='output_diffusion-sdf/UNCOND__cp20' \
|
| 18 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 19 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/zero_shot_sampling_cond_broken.sh
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3_zs.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/yuheliu/msn_lyh/zero_shot_data' \
|
| 3 |
+
dataset.name='msn_zeroshot' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/zero_shot_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=False \
|
| 9 |
+
model.use_mask_cond=True \
|
| 10 |
+
model.cond_num=1 \
|
| 11 |
+
model.use_oneplane=False \
|
| 12 |
+
model.use_triplane=False \
|
| 13 |
+
model.use_multiplane=False \
|
| 14 |
+
model.use_broken=True \
|
| 15 |
+
model.output_dir='output/ZERO_SHOT__COND1__use_broken__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND1__use_broken/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
| 18 |
+
|
sample_scripts/zero_shot_sampling_cond_multiplane.sh
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3_zs.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/yuheliu/msn_lyh/zero_shot_data' \
|
| 3 |
+
dataset.name='msn_zeroshot' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/zero_shot_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=False \
|
| 9 |
+
model.use_mask_cond=True \
|
| 10 |
+
model.cond_num=1 \
|
| 11 |
+
model.use_oneplane=False \
|
| 12 |
+
model.use_triplane=False \
|
| 13 |
+
model.use_multiplane=True \
|
| 14 |
+
model.use_broken=False \
|
| 15 |
+
model.output_dir='output/ZERO_SHOT__COND1__use_multiplane__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND1__use_multiplane/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/zero_shot_sampling_cond_oneplane.sh
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3_zs.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/yuheliu/msn_lyh/zero_shot_data' \
|
| 3 |
+
dataset.name='msn_zeroshot' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/zero_shot_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=False \
|
| 9 |
+
model.use_mask_cond=True \
|
| 10 |
+
model.cond_num=1 \
|
| 11 |
+
model.use_oneplane=True \
|
| 12 |
+
model.use_triplane=False \
|
| 13 |
+
model.use_multiplane=False \
|
| 14 |
+
model.use_broken=False \
|
| 15 |
+
model.output_dir='output/ZERO_SHOT__COND1__use_oneplane__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND1__use_oneplane/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/zero_shot_sampling_cond_triplane.sh
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3_zs.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/yuheliu/msn_lyh/zero_shot_data' \
|
| 3 |
+
dataset.name='msn_zeroshot' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/zero_shot_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=False \
|
| 9 |
+
model.use_mask_cond=True \
|
| 10 |
+
model.cond_num=1 \
|
| 11 |
+
model.use_oneplane=False \
|
| 12 |
+
model.use_triplane=True \
|
| 13 |
+
model.use_multiplane=False \
|
| 14 |
+
model.use_broken=False \
|
| 15 |
+
model.output_dir='output/ZERO_SHOT__COND1__use_triplane__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND1__use_triplane/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/zero_shot_sampling_guide_broken.sh
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3_zs.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/yuheliu/msn_lyh/zero_shot_data' \
|
| 3 |
+
dataset.name='msn_zeroshot' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/zero_shot_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=True \
|
| 9 |
+
model.use_mask_cond=False \
|
| 10 |
+
model.cond_num=0 \
|
| 11 |
+
model.use_oneplane=False \
|
| 12 |
+
model.use_triplane=False \
|
| 13 |
+
model.use_multiplane=False \
|
| 14 |
+
model.use_broken=True \
|
| 15 |
+
model.output_dir='output/ZERO_SHOT__GUIDE1__use_broken__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
| 18 |
+
|
sample_scripts/zero_shot_sampling_guide_multiplane.sh
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=0 python sample_msn_v3_zs.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/yuheliu/msn_lyh/zero_shot_data' \
|
| 3 |
+
dataset.name='msn_zeroshot' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/zero_shot_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=True \
|
| 9 |
+
model.use_mask_cond=False \
|
| 10 |
+
model.cond_num=0 \
|
| 11 |
+
model.use_oneplane=False \
|
| 12 |
+
model.use_triplane=False \
|
| 13 |
+
model.use_multiplane=True \
|
| 14 |
+
model.use_broken=False \
|
| 15 |
+
model.output_dir='output/ZERO_SHOT__GUIDE1__use_multiplane__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/zero_shot_sampling_guide_oneplane.sh
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3_zs.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/yuheliu/msn_lyh/zero_shot_data' \
|
| 3 |
+
dataset.name='msn_zeroshot' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/zero_shot_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=True \
|
| 9 |
+
model.use_mask_cond=False \
|
| 10 |
+
model.cond_num=0 \
|
| 11 |
+
model.use_oneplane=True \
|
| 12 |
+
model.use_triplane=False \
|
| 13 |
+
model.use_multiplane=False \
|
| 14 |
+
model.use_broken=False \
|
| 15 |
+
model.output_dir='output/ZERO_SHOT__GUIDE1__use_oneplane__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|
sample_scripts/zero_shot_sampling_guide_triplane.sh
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CUDA_VISIBLE_DEVICES=1 python sample_msn_v3_zs.py \
|
| 2 |
+
dataset.root_dir='/mnt/nas1/disk01/weidongguo/yuheliu/msn_lyh/zero_shot_data' \
|
| 3 |
+
dataset.name='msn_zeroshot' \
|
| 4 |
+
dataset.batch_size=12 \
|
| 5 |
+
dataset.minibatch_size=4 \
|
| 6 |
+
dataset.text_embedding_file='data_preproc/zero_shot_embedding.json' \
|
| 7 |
+
model.timesteps=50 \
|
| 8 |
+
model.use_mask_guide=True \
|
| 9 |
+
model.use_mask_cond=False \
|
| 10 |
+
model.cond_num=0 \
|
| 11 |
+
model.use_oneplane=False \
|
| 12 |
+
model.use_triplane=True \
|
| 13 |
+
model.use_multiplane=False \
|
| 14 |
+
model.use_broken=False \
|
| 15 |
+
model.output_dir='output/ZERO_SHOT__GUIDE1__use_triplane__cp20' \
|
| 16 |
+
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
|
| 17 |
+
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|