摘要：
当我们的模型训练好之后，需要将其参数（或整个模型）保存起来以便日后直接使用。pytorch提供了两种方法帮助我们快速、方便地保存训练好的模型 步骤 训练模型 保存模型 加载模型 训练模型 我们以二分类问题为例，训练一个神经网络，代码如下： import torch import torch.nn.f 阅读全文
posted @ 2021-09-13 21:43 foghorn 阅读(313) 评论(0) 推荐(0) 编辑
2021年9月13日
摘要：
当我们单独使用nn.Softmax()函数时，会显示上述警告，原因是softmax（）函数已经被弃用了，虽然程序还是可以运行成功，但是这个做法不被pytorch所赞成。这个写法在早期的pytorch版本是没有警告的，现在因为其他考虑，要加上有指明dim参数。 例如： m = nn.Softmax(d 阅读全文
posted @ 2021-09-13 20:58 foghorn 阅读(454) 评论(0) 推荐(0) 编辑
摘要：
LogSoftmax(dim=None) 对n维输入张量做softmax操作并取对数。 \(LogSoftmax(x_{i})=log(\frac{exp(x_{i})}{\sum_{j}exp(x_{j})})\) shape input(*): *代表任意的维度 output(*): 和输入维度 阅读全文
posted @ 2021-09-13 16:03 foghorn 阅读(168) 评论(0) 推荐(0) 编辑