pytorch

PyTorch训练时准确率不变的原因

2022-11-24  本文已影响0人  午字横
  1. 尝试调整学习率;
    lr=0.1 改为 lr=0.001
  2. 尝试更换一下损失函数;
  3. 尝试减少层数，增加层宽度;
  4. 尝试更换激活函数；
  5. 尝试打乱数据，调整batch_size大小
    train_loader = Data.DataLoader(dataset=dataset, batch_size=BATCH_SIZE, shuffle=True,drop_last=False)
上一篇 下一篇

猜你喜欢

热点阅读