关于接着上次epoch训练的问题

想接着某一次的epoch继续训练,修改了train.py里面的参数:

parser.add_argument(
“-w”, “–weight_file”, default=“log-of-retinanet_res50_coco_1x_800size/epoch_17.pkl”, type=str, help=“weights file”,
)

但会报错,

按照报错,将model.backbone.bottom_up.load_state_dict(weights) 修改为model.backbone.bottom_up.load_state_dict(weights, strict=False)

可以开始训练,但发现损失变成从头训练了,想请教下具体该如何操作,感谢

你代码用的是最新的1.0RC版本吧?之前检测模型的pretrain weight无法在1.0RC上使用,目前我们正在重新训练pretrain weight模型,敬请关注后续更新