想接着某一次的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)
可以开始训练,但发现损失变成从头训练了,想请教下具体该如何操作,感谢