其他分享
首页 > 其他分享> > AlexNet白板复现

AlexNet白板复现

作者:互联网

1.网络结构
AlexNet网络构成
2.一些新知学习
学到的新知识
3.遇到的问题
(1)class AlexNet(nn.Module):中,Module的M要大写;
(2)第一个全连接层nn.Linear(256*5*5, 4096)中,经计算应该是66,但是运行代码的过程中,发现55才是正确的?????
(3)训练过程中的参数更新:

optimizer.zero_grad() # 将梯度归零
l.backward() # 反向传播计算得到每个参数的梯度值
optimizer.step() # 通过梯度下降执行一步参数更新

(4)在torch.utils.data.DataLoader中,参数num_workers若为0,则代表不用额外的进程来加速读取数据。

标签:optimizer,nn,梯度,白板,Module,参数,复现,AlexNet
来源: https://blog.csdn.net/weixin_45739060/article/details/120611603