Open qiyuefun opened 1 year ago
如果在已训练好的模型上,增加其他情景的数据,然后重新训练,目前工程版本是否支持,如果支持,应该如何修改
@qiyuefun 可以把已训好的模型作为预训练参数载入进去然后在新的数据上进行迁移学习。你可以直接使用fintune部分的逻辑: https://github.com/cfzd/Ultra-Fast-Lane-Detection/blob/85eda710482eebe1a48f90316de7cfec7399bbf3/train.py#L122-L129
好的,感谢。
如果在已训练好的模型上,增加其他情景的数据,然后重新训练,目前工程版本是否支持,如果支持,应该如何修改