V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
V2EX 提问指南
Richard14
V2EX  ›  问与答

AlexNET 训练准确率奇差,可能是什么原因导致的?

  •  
  •   Richard14 · 2021-11-14 16:28:29 +08:00 · 529 次点击
    这是一个创建于 866 天前的主题,其中的信息可能已经有所发展或是发生改变。

    如题,初学过程中实现 alexnet ,原以为训练效果会挺好看,结果准确率奇差无比,可能是什么原因导致的?

    数据集因为 imagenet 太大了,用的 fasionMnist ,原格式[1,28,28],放大到[1,227,227]后输入

    网络结构写这样:

    class NeuralNetwork(nn.Module):
        def __init__(self):
            super(NeuralNetwork, self).__init__()
            self.conv2d_relu_stack = nn.Sequential(
                nn.Conv2d(1, 96, kernel_size = 11, stride=4, padding=1), nn.ReLU(),
                nn.MaxPool2d(kernel_size = 3, stride=2),
                nn.Conv2d(96, 256, kernel_size = 5, padding=2), nn.ReLU(),
                nn.MaxPool2d(kernel_size = 3, stride=2),
                nn.Conv2d(256, 384, kernel_size = 3, padding=1), nn.ReLU(),
                nn.Conv2d(384, 384, kernel_size = 3, padding=1), nn.ReLU(),
                nn.Conv2d(384, 256, kernel_size = 3, padding=1), nn.ReLU(),
                nn.MaxPool2d(kernel_size = 3, stride=2),
                nn.Flatten(),
                nn.Linear(9216, 4096), nn.ReLU(), nn.Dropout(p=0.5),
                nn.Linear(4096, 4096), nn.ReLU(), nn.Dropout(p=0.5),
                nn.Linear(4096, 10)
            )
    

    训练 50 个 epoch 后准确率 85%,个人感觉低的十分夸张,哪里写错了吗?

    完整代码: https://pastebin.com/enzCeHin

    1 条回复    2021-11-15 09:12:26 +08:00
    kilasuelika
        1
    kilasuelika  
       2021-11-15 09:12:26 +08:00 via Android
    你可以先拿现成的网络来试试看准确率多少,然后再对比源代码。
    alexnet 应该有已经封装好的实现。
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   我们的愿景   ·   实用小工具   ·   2557 人在线   最高记录 6543   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 30ms · UTC 15:42 · PVG 23:42 · LAX 08:42 · JFK 11:42
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.