Fork me on github
摘要: 这个程序由两个文件组成,一个训练脚本,一个测试脚本。安装好相应依赖环境之后即可进行训练,MNIST数据集使用torchvision.datasets.mnist包自动下载。 mnistTrain.py # -*- coding: utf-8 -*- import torch from torchvi 阅读全文
posted @ 2022-07-16 13:15 fang-d 阅读(126) 评论(1) 推荐(1) 编辑