python main.py --model="model_name" --save="model_name" --dataset="dataset_name" --input_size=image_size
python main.py --model="model_name" --save="model_name" --e="model_path"
python main.py --model="model_name" --save="model_name" --dataset="dataset_name" --input_size=image_size --resume="checkpoint_path"