mimc_rl / scripts /uncon_gen.sh
wangyanhui666's picture
fine tune decoder with mask
9cf79cf
cd .. \
python gen_img_uncond.py --temp 6.0 --num_iter 20 \
--ckpt "./ckpt_pretrained/mage-vitb-1600.pth" --batch_size 16 --num_images 32 \
--model mage_vit_base_patch16 --output_dir "./output/uncond_gen/mage-vitb/6.0" \
--gpu_id 5 --cuda \