VanillaNet
VanillaNet copied to clipboard
权重使用
老师您好,我使用net.load_state_dict(torch.load(model_weight_path, map_location='cpu'))加载模型测试时,出现RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory错误,我的路径没问题,20几个权重都是这个错误,请问是什么原因呢?期待您的解答,我是一名研三学生,在应用您的模型时出现了这个问题,期望您百忙之中抽时间为我解答疑惑,万分感谢!
你好,可能有以下几个原因:1.检查下载的权重是否损坏,尝试重新下载一份权重。2.使用的pytroch版本和requirements里需求的版本没有对应上。