![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
PyTorch
jj_千寻
qq:425776024
展开
-
PyTorch-批量训练技巧
来自:https://morvanzhou.github.io/tutorials/machine-learning/torch/3-05-train-on-batch/ import torch import torch.utils.data as Data torch.manual_seed(1) BATCH_SIZE = 8 # 批训练的数据个数 x = torch.linsp...转载 2019-03-06 16:11:21 · 1953 阅读 · 1 评论 -
PyTorch-模型保存与加载
保存: model = LinearRegression() # ......各种操作 model.eval() #训练完成,保存状态字典到linear.pkl torch.save(model.state_dict(), './linear.pkl') 加载: model = LinearRegression() model.load_state_dict(torch.load('l...原创 2019-03-06 16:15:46 · 2004 阅读 · 1 评论 -
DL-深度神经网络原理推导及PyTorch实现
1.前向传播 引用一个网站的图: 具体来说,就是2行代码,图片中的f为激活函数,这里用sigmoid作为激活函数,事实上有很多其它的套路,这里只讲神经网络的数学原理及初级使用,不会做任何深入扩展: def feedforward(self, a): # a:input for b, w in zip(self.biases, self.weig...原创 2019-03-06 15:40:03 · 1163 阅读 · 0 评论