- 博客(3)
- 收藏
- 关注
RSS订阅原创 使用torch.load()加载模型参数时，提示“xxx.pt is a zip archive(did you mean to use torch.jit.load()?)“
因为训练的pytorch版本是0.4的，测试的pytorch版本是1.2的，所以在测试的时候报错：解决方法：新建一个.py文件：写入如下代码：import torchstate_dict = torch.load('.pth路径名')torch.save(state_dict, ".pth路径名", _use_new_zipfile_serialization=False)将代码在原来的pytorch版本下面运行，将文件转换成非.zip文件，然后就可以在现在的pytorch版本下加载了
2021-10-09 13:21:02
375
原创 AttributeError NoneType object has no attribute data
AttributeError NoneType object has no attribute data我在自己模型中加注意力的时候出现的这个问题，上网查询这个问题，大部分回答都是模型中定义的某个层没有参与到前向传播，所以反向传播，计算loss时，grad is None。主要是要找到未参与计算的层，并注释掉。因为我只加了一个注意力，所以把注意力里面的所有参数requires_grad都置为False。具体操作如下：...
2021-08-30 16:49:08
1303
原创 训练数据集的时候GPU老是lost
在用3个GPU跑数据集的时候，总是跑了不到三分钟吧就会出现Unable to determine the device handle for GPU 0000:83:00.0: GPU is lost.Reboot the system to recover，但是如果一个GPU跑的话就可以正常运行，我在网上看了很多解决方法，大多都是说显卡驱动和cuda的版本不对，但是我都试过了发现还是不对，最后发现是有一个显卡没有跟其他显卡连接起来，如图所示，连接上去之后就可以正常运行了，提供一个解决思路。...
2021-04-16 11:23:53
2428
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人