search

cfzd / Ultra-Fast-Lane-Detection

Ultra Fast Structure-aware Deep Lane Detection (ECCV 2020)
MIT License
1.76k stars 491 forks source link

how to get the weights without seg weight when use_aux is True? #327

Open kidcats opened 1 year ago

kidcats commented 1 year ago

i train you code with use_aux is True,but the weight is 800M,how to save the weight as 250W?

cfzd commented 1 year ago

@kidcats The weight contains the state of model, optimizer. You can only save the state of the model. https://github.com/cfzd/Ultra-Fast-Lane-Detection/blob/353df107756b8c03c22c27201e33fc63d84ecfe6/utils/common.py#L66