一个经典的Pytorch神经网络分类模型训练框架