当前位置：网站首页>ValueError: loaded state dict contains a parameter group that doesn‘t match the size of optimizer‘s

ValueError: loaded state dict contains a parameter group that doesn‘t match the size of optimizer‘s

2022-04-23 05:42:00 【fattt_】

解决训练时报错：ValueError: loaded state dict contains a parameter group that doesn’t match the size of optimizer’s group
大致意思：加载状态 dict 包含一个与优化器组的大小不匹配的参数组
暂时解决：直接注释掉 optimizer.load_state_dict(ckpt[‘optimizer’])
因为
if cfg.BACKBONE.TRAIN_EPOCH == epoch:
logger.info(‘start training backbone.’)
optimizer, lr_scheduler = build_opt_lr(model.module, epoch)
logger.info(“model\n{}”.format(describe(model.module)))
当epoch=10 的时候，optimizer会重新生成。

版权声明
本文为[fattt_]所创，转载请带上原文链接，感谢
https://blog.csdn.net/weixin_41332732/article/details/114395095

边栏推荐

猜你喜欢

随机推荐