Open MingFengHill opened 6 months ago
我也遇到了这个问题,这个最后的错误是:RuntimeError: ckpts/pretrain_sed_net_InstBest.pth is a zip archive (did you mean to use torch.jit.load()?),好像是预训练模型格式的问题。请问您解决这个问题了吗?
这个是pytorch版本的问题。 在pytorch1.6版本下运行: state_dict = torch.load("xxx.pth") torch.save(state_dict, "xxx.pth", _use_new_zipfile_serialization=False) 则将权重文件转换成非zip格式。
这个是pytorch版本的问题。 在pytorch1.6版本下运行: state_dict = torch.load("xxx.pth") torch.save(state_dict, "xxx.pth", _use_new_zipfile_serialization=False) 则将权重文件转换成非zip格式。
Thanks, it works.
I configured the python environment according to ParSeNet's readme. But when I execute the test code I get the following error:
This seems to be caused by the wrong pytorch version, would it be possible to provide
environment.yml
?