PYTORCH保存模型时报错AttributeError: Can‘t pickle local object ‘Stage.__init__.<locals>.<lambda>‘

解决方法:

点开报错的代码位置,将代码里面的import pickle改成import dill as pickle

猜你喜欢

转载自blog.csdn.net/m0_48095841/article/details/129447709
今日推荐