Pytorch学习纪要[ongoing]

本文推荐莫烦Python教程作为初学者资源,并详细介绍了使用PyTorch搭建神经网络的方法,包括自定义类继承、使用Sequential快速搭建、保存模型、Batch Training、Dropout层的使用与切换、Batch Normalization的实现及其在训练与预测阶段的作用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

莫烦python教程和视频是不错的入门资源:https://morvanzhou.github.io/

  1. 搭建神经网络,可以自建class继承torch.nn.Module,也可以使用nn.Sequential快速搭建
  2. 使用torch.save(net.state_dict,'file.pkl') 比torch.save(net,'file.pkl')会稍快一些,实测生成的文件前者确实较小。
  3. Batch training: ds=torch.utils.data.TensorDataset(x,y); data.DataLoader(batch_size=10, shuffle=True, dataset=ds)
  4. Dropout layer. 实现上只需加入dropout层即可,需要注意的是,在训练结束后,需要predict时,需要调用net.eval()切换模式禁用dropout
  5. Batch normalization. 在每一层都加入一个normalization的操作使得,信息能够很好地传递到网络深层。否则的话,多大或过小的x值在经过激活函数时的效果与足够大和足够小的值没有区分度,导致信息的丢失。另外,由于这一normalizaition的操作中用到的均值和variance估计是有偏的,于是叠加了scale和偏移量两个参数,在训练时被更新。在训练结束后也需要调用net.eval()切换到调用模式,否则上述两个参数会根据新一次的mean和variance去调整。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值