- 博客(6)
- 收藏
- 关注
RSS订阅转载 基于pytorch得预训练模型
Resnet:model_urls = {undefined 'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth', 'resnet34': 'https://download.pytorch.org/models/resnet34-333f7ec4.pth', 'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e35...
2022-01-14 17:27:11
140
原创 Pyinstaller 打包pytorch时遇到得问题
在打包pytorch项目时遇到了torch打包不进去得问题（好像是），exe打开之后闪退，报错如下仔细查看打包时得打印信息，发现有一句lib not found torch_python.dll *******，考虑是这句话得原因，但是查自己打包得路径下是有这个dll库得，会不会是没有找到自己环境中得这个dll，最后通过在打包时把这个路径链接进来，解决了问题！！具体命令为 pyinstaller --paths 你没找到得dll路径，比如我得在**torch/lib下 ***.py...
2021-11-30 11:38:25
2441
3
原创 RuntimeError: Trying to backward through the graph a second time, but the buffers have already been
遇到的问题记录：这个错误的意思是在进行反向传播时，前面缓冲区里的计算图已经被释放了。在前向传播过程中每进行一次计算都会被加入到缓冲区的计算图中，而反向传播时，根据计算图中需要进行梯度传递的部分进行反向梯度计算。我遇到的这个问题是在同时迭代更新两个网络的过程。当第一个网络进行backward之后，相应的计算图就已经被释放了，而两个网络的训练损失有部分重叠，所以当第二个网络backward时，就会出现这个问题。解决的方法是将重叠部分的损失进行重新计算用于第二个网络。注意两个网络的损失要分开计算，在计算过
2021-10-05 15:52:26
253
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人