gwd200 commited on
Commit
854da5a
·
verified ·
1 Parent(s): a19bce9

Upload 28 files

Browse files
Files changed (28) hide show
  1. sample_scripts/sampling_cond_broken.sh +18 -0
  2. sample_scripts/sampling_cond_multiplane.sh +23 -0
  3. sample_scripts/sampling_cond_multiplane_broken.sh +17 -0
  4. sample_scripts/sampling_cond_oneplane.sh +24 -0
  5. sample_scripts/sampling_cond_random.sh +17 -0
  6. sample_scripts/sampling_cond_random_eval_only.sh +17 -0
  7. sample_scripts/sampling_cond_triplane.sh +25 -0
  8. sample_scripts/sampling_cond_triplane_broken.sh +23 -0
  9. sample_scripts/sampling_cond_triplane_multiplane.sh +25 -0
  10. sample_scripts/sampling_cond_triplane_multiplane_broken.sh +23 -0
  11. sample_scripts/sampling_guide_broken.sh +18 -0
  12. sample_scripts/sampling_guide_multiplane.sh +23 -0
  13. sample_scripts/sampling_guide_multiplane_broken.sh +23 -0
  14. sample_scripts/sampling_guide_oneplane.sh +17 -0
  15. sample_scripts/sampling_guide_random.sh +17 -0
  16. sample_scripts/sampling_guide_triplane.sh +23 -0
  17. sample_scripts/sampling_guide_triplane_broken.sh +23 -0
  18. sample_scripts/sampling_guide_triplane_multiplane.sh +23 -0
  19. sample_scripts/sampling_guide_triplane_multiplane_broken.sh +23 -0
  20. sample_scripts/sampling_uncond.sh +19 -0
  21. sample_scripts/zero_shot_sampling_cond_broken.sh +18 -0
  22. sample_scripts/zero_shot_sampling_cond_multiplane.sh +17 -0
  23. sample_scripts/zero_shot_sampling_cond_oneplane.sh +17 -0
  24. sample_scripts/zero_shot_sampling_cond_triplane.sh +17 -0
  25. sample_scripts/zero_shot_sampling_guide_broken.sh +18 -0
  26. sample_scripts/zero_shot_sampling_guide_multiplane.sh +17 -0
  27. sample_scripts/zero_shot_sampling_guide_oneplane.sh +17 -0
  28. 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'