search

hirotomusiker / CLRerNet

The official implementation of "CLRerNet: Improving Confidence of Lane Detection with LaneIoU"
Apache License 2.0
181 stars 19 forks source link

--resume-from #26

Closed GH-W5 closed 10 months ago

GH-W5 commented 11 months ago

unexpected key in source state_dict: ema_backbone_model_base_layer_0_weight, ema_backbone_model_base_layer_1_weight,... why?

hirotomusiker commented 11 months ago

Could you paste your training command? Probably you resumed from clrernet_culane_dla34_ema.pth.

GH-W5 commented 11 months ago

/home/ttt/anaconda3/envs/mmseg_cd/bin/python /disk_sda/abc/Lane_Detection_project/CLRerNet/tools/train.py configs/clrernet/culane/clrernet_culane_dla34_ema.py --resume-from /disk_sda/CULane/run_work_dirs/clrernet_beijing_dla34_ema/epoch_1.pth

hirotomusiker commented 11 months ago

Thanks, please check out the Resume training with ExpMomentumEMAHook part in https://mmdetection.readthedocs.io/en/dev/faq.html . You need to resume the EMA weights additionally.

Adding --cfg-options custom_hooks.0.resume_from={your_pth_file} should do the trick.

hirotomusiker commented 10 months ago

We close this issue as resolved.