pth模型是python3 的 所以我用
param = torch.load(pthfile)加载了模型 把处理后的数据存为npz文件
但是在python2环境下
weights = np.load(weights_dir ,allow_pickle=True)['arr_0'].item()报错
请问有什么办法能解决 python版本的兼容性问题呢?
pth模型是python3 的 所以我用
param = torch.load(pthfile)加载了模型 把处理后的数据存为npz文件
但是在python2环境下
weights = np.load(weights_dir ,allow_pickle=True)['arr_0'].item()报错
请问有什么办法能解决 python版本的兼容性问题呢?