Fix

ZHEQIUSHUI · Jul 12, 2022 · 699068a · 699068a
1 parent e124601
commit 699068a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolov6/core/evaler.py b/yolov6/core/evaler.py
@@ -322,7 +322,7 @@ def convert_to_coco_format(nums, boxes, scores, classes, paths, shapes, ids):
             return pred_results
 
         context, bindings, binding_addrs, trt_batchsize = init_engine(engine)
-        assert trt_batchsize >= self.batch_size, f'--batchsize {self.batch_size} must <= tensorrt batchsize {trt_batchsize}'
+        assert trt_batchsize >= self.batch_size, f'--batch {self.batch_size} must <= tensorrt batchsize {trt_batchsize}'
         tmp = torch.randn(self.batch_size,3,self.img_size,self.img_size).to(self.device)
         # warm up for 10 times
         for _ in range(10):