| CUDA_VISIBLE_DEVICES=1 python sample_msn_v3_zs.py \ | |
| dataset.root_dir='/mnt/nas1/disk01/weidongguo/yuheliu/msn_lyh/zero_shot_data' \ | |
| dataset.name='msn_zeroshot' \ | |
| dataset.batch_size=12 \ | |
| dataset.minibatch_size=4 \ | |
| dataset.text_embedding_file='data_preproc/zero_shot_embedding.json' \ | |
| model.timesteps=50 \ | |
| model.use_mask_guide=False \ | |
| model.use_mask_cond=True \ | |
| model.cond_num=1 \ | |
| model.use_oneplane=True \ | |
| model.use_triplane=False \ | |
| model.use_multiplane=False \ | |
| model.use_broken=False \ | |
| model.output_dir='output/ZERO_SHOT__COND1__use_oneplane__cp20' \ | |
| model.checkpoint_for_sample='EXP/COND/LR0.001__BS24__TS50__COND1__use_oneplane/model-20.pt' \ | |
| vae.pvqvae_ckpt='pretrained_models/pvqvae.pth' | |