多层感知机的简洁实现

Sigmoid:
Sigmoid

三层网络+ReLU
和原双层对比在初始训练时精度较低,后逐渐趋于一致

image
还有比我更牛逼的吗?

调整了层数和batch_size,代码和结果如下:

import torch

from torch import nn

from d2l import torch as d2l

net = nn.Sequential(nn.Flatten(),nn.Linear(784,512),nn.ReLU(),nn.Linear(512,256),

                nn.ReLU(),nn.Linear(256,64),nn.ReLU(),nn.Linear(64,10))

def init_weights(m):

if type(m) == nn.Linear:

nn.init.normal_(m.weight,std=0.01)

net.apply(init_weights);

batch_size,lr,num_epochs = 64,0.1,10

loss = nn.CrossEntropyLoss(reduction=‘none’)

trainer = torch.optim.SGD(net.parameters(),lr=lr)

train_iter,test_iter = d2l.load_data_fashion_mnist(batch_size)

d2l.train_ch3(net,train_iter,test_iter,loss,num_epochs,trainer)

image

均值初始化时,需要将a b范围设置为-1/sqrt(n) 和 1/sqrt(n),n是每层的权重个数。

nn.init.uniform_(m.weight, a = -1/len(m.weight) ** (1/2), b=1/len(m.weight) ** (1/2) )

def init_weights(m):
if type(m) == nn.Linear:
# nn.init.normal_(m.weight , mean = 0 ,std = 0.01) # 正态分布
nn.init.uniform_(m.weight, a = -1/len(m.weight) ** (1/2), b=1/len(m.weight) ** (1/2) ) # 均匀分布
# nn.init.eye_(m.weight) # 填充1 使得特征尽可能多的保留
# nn.init.dirac_(m.weight) # 用dirac 函数填充 ,尽量保持卷积层中的特征
可以试试填充1的初始化权重 效果也挺好

1 Like

小白想问一下,运行4.3节的时候,怎么会报错,怎么修改。。。

是的,从源码看您可以通过改变Animator类的ylim参数来改变y轴的显示范围

你确定是sigmoid更好么
为什么我看这个图最后的 train_acc 和 test_acc 都比前两个更低啊 , 明显Tanh更好啊

1 Like

如果出现过拟合的话,该怎么解释 train_acc 的降低和 train_loss 的增大呢?

Original:

Screenshot 2021-12-15 172110_tanh

升级了,反而#@save保存不到d2l是因为什么,现在是1.0.3

TANH:
image
SIGMOID:
image
RELU:
image

你好,请问你的trian loss 是怎么显示出来的,为什么我的损失图像没有trian loss,即使调整了显示的范围,是因为loss太大了吗
image

是太小了
image
我把loss打印出来看了