你换了模型跑,很可能就会出现这个问题,简单粗暴的解决办法就是重写save_checkpoint和load_checkpoint,也就是检查点重新放一个文件夹
pytorch模型加载出现报错:raise RuntimeError(‘Error(s) in loading state_dict for {}:\n\t{}‘.format(RuntimeErr
最新推荐文章于 2024-07-12 11:12:53 发布
你换了模型跑,很可能就会出现这个问题,简单粗暴的解决办法就是重写save_checkpoint和load_checkpoint,也就是检查点重新放一个文件夹