报错信息:
C:\ProgramData\Anaconda3\envs\ywx_pytorch_env\lib\site-packages\torch\serialization.py:453: SourceChangeWarning: source code of class '__main__.CNN' has changed. you can retrieve the original source code by accessing the object's source attribute or set `torch.nn.Module.dump_patches = True` and use the patch tool to revert the changes.
warnings.warn(msg, SourceChangeWarning)
原因分析:
pytorch模型预测的时候出现这个错误,原来是修改了CNN网络代码里面的一个注释,导致与最开始写的CNN网络代码不完全一样,因此出现了这个警告信息,注释都不能改。
解决方案:
恢复原CNN代码的任何修改,需要和原CNN代码一模一样,包括注释。