![](https://img-blog.csdnimg.cn/20190918140145169.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
pytorch笔记
文章平均质量分 83
pytorch笔记
wwweiyx
这个作者很懒,什么都没留下…
展开
-
pytorch加载模型错误 Missing key(s) RuntimeError: Error(s) in loading state_dict for 多卡加载错误
pytorch 加载模型错误:RuntimeError: Error使用 torch.save() 保存权重时,通过 model.module.state_dict() 获取模型权重。包装后的模型参数的关键字会比没用 nn.DataParallel 包装的模型参数的关键字前面多一个"在使用nn.DataParallel之前就先读取模型,然后再使用nn.DataParallel。加载模型时使用 nn.DataParallel。保存权重前增加 module。把 module. 删掉。手动添加 module.原创 2022-08-11 17:56:28 · 1246 阅读 · 0 评论 -
[模块]pytorch模型的储存和载入
pytorch保存和载入模型原创 2022-07-07 12:38:56 · 1075 阅读 · 0 评论 -
pytorch 中的 .detach() .clone()
pytorch tensor 中的 和 detach() 的用法在写代码时经常能见到通过 操作生成一个和原本 tensor 值相同的新 tensor为什么需要同时使用 和 ,接下来通过代码进行说明生成两个 tensor,并且求梯度输出结果:可以看到 a, b 的梯度分别为 [2., 2.],[1., 1.]使用 a_=a.detch() 脱离计算图在上面的代码中加上 并且使用 计算和 backward()输出结果:此时 a 的梯度为 none,因为 生成了一个新的 tensor 并且从计算图原创 2022-06-06 17:32:06 · 1279 阅读 · 0 评论 -
pytorch中常见的损失函数
pytorch中常见的损失函数参考自:https://mp.weixin.qq.com/s/2oUNYUwkrVUN1fV4zDER7Q文章目录pytorch中常见的损失函数1.交叉熵损失交叉熵简介BCELossBCEWithLogitsLossNLLLossCrossEntropyLoss2.KL散度KL散度简介KLDivLoss3.平均绝对误差(L1范数)L1范数简介L1Lossl1_loss04 均方误差损失(L2范数)L2范数简介MSELossmse_loss5.铰链损失函数Hinge loss原创 2021-03-13 20:45:17 · 1212 阅读 · 0 评论 -
pytorch中tensor创建方法
tensor的创建方法总结直接创建code作用torch.tensor()直接创建torch.from_numpy(ndarray)从numpy创建tensor按值创建code作用torch.tensor()直接创建torch.from_numpy(ndarray)从numpy创建tensortorch.zeros() torch.zeros_like()创建全零张量torch.ones() torch.ones_原创 2020-06-09 18:47:59 · 3753 阅读 · 0 评论