python main.py \
--feature_extractor inception_v3 \
--fid_model_name inception_v3 \
--fid_dataset_name brain \
--plot_images False \
--gen_class_cond True \
--count_threshold 4.0 \
--noise_multiplier 2.1793 \
--model_path models/imagenet64_cond_270M_250K.pt \
--lookahead_degree 8 \
--image_size 64x64 \
--private_image_size 64 \
--data_folder data/brain_train \
--num_samples_schedule 6000,6000,6000,6000,6000,6000,6000,6000,6000,6000,6000,6000,6000,6000,6000,6000,6000,6000,6000,6000,6000 \
--variation_degree_schedule 0,2,4,6,8,10,12,14,16,18,20,22,24,26,28,30,32,34,36,40,42 \
--num_fid_samples 3000 \
--num_private_samples 5700 \
--make_fid_stats True \
--result_folder result_brain_conditional_4 \
--tmp_folder tmp_result_brain_conditional \
--api improved_diffusion \
--model_image_size 64 \
--num_channels 192 \
--num_res_blocks 3 \
--learn_sigma True \
--class_cond True \
--diffusion_steps 4000 \
--noise_schedule cosine \
--rescale_learned_sigmas False \
--rescale_timesteps False \
--use_ddim True \
--timestep_respacing ddim10 \
--batch_size 500 \
--data_loading_batch_size 500 \
--feature_extractor_batch_size 500 \
--fid_batch_size 500