Pytorch
shimmer_ghq
这个作者很懒,什么都没留下…
展开
-
Pytorch训练时GPU利用率为0或很低
pytorch训练时,显存明显上去了,模型和数据都在gpu上,但GPU利用率仍为0%原创 2022-07-10 20:22:58 · 5804 阅读 · 0 评论 -
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
添加代码:torch.backends.cudnn.enabled=False原创 2022-03-02 18:56:55 · 156 阅读 · 0 评论 -
pytorch加载模型时出现.....xxx.pth is a zip archive (did you mean to use torch.jit.load()?)
pytorch加载模型时出现…xxx.pth is a zip archive (did you mean to use torch.jit.load()?)这是因为新版本pytorch默认把模型存储为压缩文件,老版本不是压缩文件解决方法:在新版本中将模型加载后,再存储为非压缩文件,再用老版本加载:import torchstate_dict = torch.load("xxx.pth") #xxx.pth或者xxx.pt就是你想改掉的权重文件torch.save(state_dict, "xxx原创 2022-02-27 15:13:42 · 4974 阅读 · 4 评论 -
【Pytorch】unsqueeze()与squeeze()
squeeze()官方文档简而言之,squeeze()用于压缩维度,unsqueeze()用于扩充维度Pytorch中将图像输入CNN进行预测之前,进行unsqueeze操作:img=torch.unsqueeze(img, dim=0)是因为网络的接收输入是一个mini-batch,image unsqueeze后第一个维度是留给batch size的...原创 2022-02-23 14:45:31 · 1286 阅读 · 0 评论