千寻的博客

来自：https://morvanzhou.github.io/tutorials/machine-learning/torch/3-05-train-on-batch/import torchimport torch.utils.data as Datatorch.manual_seed(1)BATCH_SIZE = 8 # 批训练的数据个数x = torch.linsp...

保存：model = LinearRegression()# ......各种操作model.eval()#训练完成，保存状态字典到linear.pkltorch.save(model.state_dict(), './linear.pkl')加载：model = LinearRegression()model.load_state_dict(torch.load('l...

1.前向传播引用一个网站的图：具体来说，就是2行代码，图片中的f为激活函数，这里用sigmoid作为激活函数，事实上有很多其它的套路，这里只讲神经网络的数学原理及初级使用，不会做任何深入扩展：def feedforward(self, a): # a:input for b, w in zip(self.biases, self.weig...