摘要: 1 import torch 2 from torch import optim,nn 3 import visdom 4 import torchvision 5 from torch.utils.data import DataLoader 6 7 from pokemon import Pok 阅读全文
posted @ 2020-02-19 15:38 一大碗小米粥 阅读(881) 评论(0) 推荐(0) 编辑
摘要: 1 import torch 2 import os,glob 3 import random,csv 4 5 from torch.utils.data import Dataset,DataLoader 6 7 from torchvision import transforms 8 from 阅读全文
posted @ 2020-02-19 15:36 一大碗小米粥 阅读(1187) 评论(1) 推荐(1) 编辑
摘要: 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) 编辑
摘要: 1 import torch 2 from torch import nn 3 from torch.nn import functional as F 4 from torch import optim 5 6 import torchvision 7 from matplotlib import 阅读全文
posted @ 2020-02-10 20:07 一大碗小米粥 阅读(1219) 评论(1) 推荐(1) 编辑