当前位置：网站首页>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:56:00 【fattt_】

Solve training errors ：ValueError: loaded state dict contains a parameter group that doesn’t match the size of optimizer’s group
What do you mean ： Loading state dict Contains a parameter group that does not match the size of the optimizer Group
Temporary settlement ： Comment out optimizer.load_state_dict(ckpt[‘optimizer’])
because
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)))
When epoch=10 When ,optimizer Will regenerate .

版权声明
本文为[fattt_]所创，转载请带上原文链接，感谢
https://yzsam.com/2022/04/202204230541585097.html

边栏推荐

猜你喜欢

随机推荐