训练模型
训练
参考文档的资料训练一个模型
说明
训练算力要求较高,最好是有nvidia的gpu,我自己测试过在3070 laptop上大概需要十多个小时的时间,使用V100可以缩短大概一半
如果你的电脑没有独显,请先尝试让它在CPU上跑起来,然后再联系学长使用云算力
实验室经费有限,请尽量减少使用时间
量化
模型训练通常是在fp32(32位浮点数,即单精度)或者fp16(16位浮点数,即半精度)上进行的,在比较高精度训练的模型虽然效果可能好但性能开销也大
我们可以通过量化这个步骤,降低模型的大小和计算量,便于运行在更小的设备上
一般认为,精度从fp32降到fp16不会对效果产生明显影响,量化到int8或者fp8时效果仍可以接受,但int4甚至更低的就没有意义了
ultralytics库提供了一个工具,可以在格式转换时便捷地进行模型量化等操作,点此查看文档
任务
完成任务13