--image_size
--model_name
--save_dir
--ckpt_nmbr