python train.py --data_path "./datasets/flowers" --num_classes 5 --input_size 256 --gpu 0