search

yelusaleng / RRU-Net

Official repository for "RRU-Net: The Ringed Residual U-Net for Image Splicing Forgery Detection" (CVPRW 2019)
107 stars 18 forks source link

关于模型推理报错(predict.py) #37

Open RealWaltz opened 1 year ago

RealWaltz commented 1 year ago

您好，我在尝试运行项目中的predict.py时遇到了一些问题：

项目环境是：python3.7, torch1.0.0,使用best_model.pth作为输入模型 运行后遇到报错(指定cpu或者cuda均会报同一错误)：

File "predict.py", line 67, in net.cuda() File "/root/miniconda3/envs/RRU2/lib/python3.7/site-packages/torch/serialization.py", line 367, in load return _load(f, map_location, pickle_module) File "/root/miniconda3/envs/RRU2/lib/python3.7/site-packages/torch/serialization.py", line 528, in _load magic_number = pickle_module.load(f) _pickle.UnpicklingError: A load persistent id instruction was encountered, but no persistent_load function was specified.

请问您知道这个报错的解决方案吗？十分感谢🙇‍