pytorch_train.py

本代码为python在pytorch上代码，需要建立文件夹，其子文件夹命名为train和val，分别在子文件夹中放入训练和测试数据，改变代码中数据集地址即可实现分类