摘要: main dataloader train test 相对LeNet5的主函数来讲,仅仅是更换了模型名称,其他部分没有变化。 import torch from torch.utils.data import DataLoader from torchvision import datasets f 阅读全文
posted @ 2020-02-16 20:06 一大碗小米粥 阅读(1804) 评论(0) 推荐(0) 编辑
摘要: LeNet5网络和CIFAR10数据集 main函数--dataloader--train--test 1 import torch 2 from torch.utils.data import DataLoader 3 from torchvision import datasets 4 from 阅读全文
posted @ 2020-02-16 16:51 一大碗小米粥 阅读(599) 评论(0) 推荐(0) 编辑