pytorch构建LeNet模型

最新推荐文章于 2024-05-14 10:21:54 发布

VIP文章「已注销」

最新推荐文章于 2024-05-14 10:21:54 发布

阅读量157

点赞数

分类专栏：深度学习文章标签： pytorch 深度学习人工智能

本文链接：https://blog.csdn.net/m0_73817310/article/details/127149655

版权

LeNet模型

import torch
from torch import nn
import torch 
print(torch.__version__)

1.11.0+cpu

构建模型

class LeNet(nn.Module):
    def __init__(self):
        super(LeNet, self).__init__()
        self.conv = nn.Sequential(
            nn.Conv2d(1, 6, 5), # in_channels, out_channels, kernel_size
            nn.Sigmoid(),
            nn.MaxPool2d(2, 2), # kernel_size, stride
            nn.Conv2d(6, 16, 5),
            nn.Sigmoid(),
            nn.MaxPool2d(2, 2)
        )
        self.fc = nn.Sequential(
            nn.Linear(16*4*4, 120),
            nn.Sigmoid(),
            nn.Linear(120, 84),
            nn.Sigmoid(),
            nn.Linear(84, 10)
        )
    def forward(self, img):
        feature = self.conv(img)
        out