File size: 743 Bytes
854da5a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | 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=True \
model.use_mask_cond=False \
model.cond_num=0 \
model.use_oneplane=False \
model.use_triplane=False \
model.use_multiplane=False \
model.use_broken=True \
model.output_dir='output/ZERO_SHOT__GUIDE1__use_broken__cp20' \
model.checkpoint_for_sample='EXP/RAW4GUIDE/LR0.001__BS24__TS50/model-20.pt' \
vae.pvqvae_ckpt='pretrained_models/pvqvae.pth'
|