我开始转换的时候设置的batchsize=1,是静态维度,模型上线时转报错信息如下: 2022-09-29T02:53:42.192Z [34mINFO[0m execute/mosn.go:43 [tail][/opt/tritonserver.log]: E0929 02:53:41.941147 83 model_repository_manager.cc:1152] failed to load 'yolov5_obb_det_640' version 20220928215625: Invalid...
没问题,放心大胆上,我英特尔核显跑六个小时跑完了,你这个肯定没问题,我是设置的8,你可以适当加大...
只是学习性质的话,可以拿yolov5-s玩一玩,input改到320*320或者更低,应该可以整个4-8的bs。拿来做...
我batch-size设置为4,epoch为50,数据集总共一万多的照片跑一下九个多小时,虽然慢但是最后训练出来也可以用