类目参数

取值

Epoch

300

image_size

224

batch_size

32

损失函数

CrossEntropyLoss()

优化器

Adam()

learning rate

0.0001；step_size = 5；gamma = 0.5