pytorch 学习笔记 part9 LeNet 模型

通过Sequential类来实现LeNet模型 #import import sys sys.path.append(/home/kesci/input) import d2lzh1981 as d2l import torch import torch.nn as nn import torch.optim as optim import time #net class Flatten(torch.nn.Module): #展平操作 def forward(self, x): return x.view(x.shape[0], -1) class Reshap