Skip to content

训练模型

训练

参考文档的资料训练一个模型

说明

训练算力要求较高,最好是有nvidia的gpu,我自己测试过在3070 laptop上大概需要十多个小时的时间,使用V100可以缩短大概一半

如果你的电脑没有独显,请先尝试让它在CPU上跑起来,然后再联系学长使用云算力

实验室经费有限,请尽量减少使用时间

量化

模型训练通常是在fp32(32位浮点数,即单精度)或者fp16(16位浮点数,即半精度)上进行的,在比较高精度训练的模型虽然效果可能好但性能开销也大

我们可以通过量化这个步骤,降低模型的大小和计算量,便于运行在更小的设备上

一般认为,精度从fp32降到fp16不会对效果产生明显影响,量化到int8或者fp8时效果仍可以接受,但int4甚至更低的就没有意义了

ultralytics库提供了一个工具,可以在格式转换时便捷地进行模型量化等操作,点此查看文档

任务

完成任务13