文章首发及后续更新:https://mwhls.top/4048.html,无图/无目录/格式错误/更多相关请至首发页查看。
新的更新内容请到mwhls.top查看。
欢迎提出任何疑问及批评,非常感谢!
--resume-from
来加载预训练权重。my_model_config.py
的 backbone 中添加 init_cfg
,如下:backbone=dict(
type='MobileNetV2',
init_cfg=dict(
type="Pretrained",
checkpoint="pretrain\yolov3_mobilenetv2_320_300e_coco_20210719_215349-d18dff72.pth",
prefix="backbone"
)),
--resume-from
是让模型继续在该 epoch 后继续训练的,我使用的 ssdlite 的权重,它是训练 120 epoch 后的权重文件,当我恢复断点继续训练时,它会从 121 epoch 开始,而配置文件的 max_epcoh
同样也是 120,因此直接完成了训练。mmdet
的参数没提供 --load-from
,所以我就按 --resume-from
加载了,气死,一个下午没跑通怀疑自己了。--load-from
处理,但是为什么不写进配置里面,可恶。ValueError need at least one array to concatenate
的,该死,见这里。