anime-colorization / test_danbooru_cascade.sh
HighCWu's picture
First model version
052c05b
raw
history blame contribute delete
No virus
685 Bytes
MODEL_FLAGS="--image_size 32 --small_size 32 --large_size 128 --guide_size 128 --num_channels 128 --num_channels2 64 --num_res_blocks 3 --learn_sigma True --dropout 0.0 --use_attention2 False"
DIFFUSION_FLAGS="--diffusion_steps 4000 --noise_schedule cosine"
TEST_FLAGS="--batch_size 4 --seed 233"
OPENAI_LOGDIR="./danbooru2017_guided_cascaded_test_log" python scripts/cascaded_pixel_guide_sample.py --data_dir data/danbooru2017/anime --guide_dir data/danbooru2017/anime_sketch --timestep_respacing ddim25 --use_ddim True --model_path danbooru2017_guided_log/ema_0.9999_360000.pt --model_path2 danbooru2017_guided_sr_log/ema_0.9999_360000.pt $MODEL_FLAGS $DIFFUSION_FLAGS $TEST_FLAGS