项目场景:
跑模型的时候,pytorch加载模型的时候报错。由于模型是师姐给我的,我也不知道是啥结构。
问题描述
这个报错网上都搜不到,wrn不知道是啥,pip install和conda install都装不上
原因分析:
原来wrn 是 Wide Residual Networks 缩写 广度残差网络
原文连接 https://arxiv.org/abs/1605.07146
解决方案:
WideResNet-pytorch/wideresnet.py at master · xternalz/WideResNet-pytorch · GitHub
在代码目录下新建wrn文件夹,上面链接下载的wideresnet.py文件放进去,init文件
from .wideresnet import *