Lenet代码 pytorch官方demo(Lenet)
下面是LeNet的模型代码 :model.pyimport torch.nn as nnimport torch.nn.functional as Fclass LeNet(nn.Module):#继承nn.Module def __init__(self): super(LeNet, self).__init__() self.conv1 = nn.Conv2d(3, 16, 5)#3通道的输入,16卷积核,5x5的卷积核 self.poo
原创
2022-05-15 15:34:51 ·
1135 阅读 ·
0 评论