所在位置:
首页 >
营销资讯 >
行业动态 > model.load_state_dict(torch.load('best_model.pth'))
model.load_state_dict(torch.load('best_model.pth'))
时间:2022-03-16 16:00:01 | 来源:行业动态
时间:2022-03-16 16:00:01 来源:行业动态
这种做法的好处是,未来面向更多不同分道物的时候,可以不断往上叠加以增加Jetbot适合的使用场景。
例如第一次使用红锥做分道物,进行数据采集并训练出best_model_cones_1.pth,等后面用要使用定位胶带做分道物的时候,就能在best_model_cones_1.pth的基础上去训练能识别红锥与定位胶带两种分道物的best_model_cones_2.pth模型,真正发挥迁移学习的效果。
-