pytorch_train.py

【文件属性】：

文件名称：pytorch_train.py

文件大小：5KB

文件格式：PY

更新时间：2023-05-09 08:49:00

pytorch 图片分类 深度学习 python 自定义数据集

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