diff --git a/test_tipc/configs/dino/dino_r50_4scale_1x_coco_train_infer_python.txt b/test_tipc/configs/dino/dino_r50_4scale_1x_coco_train_infer_python.txt index 6d77fd46e27c3c916a4d18f32a43098d20509f9c..87117a4cc1fa4450a249d8144fd7d038c2901825 100644 --- a/test_tipc/configs/dino/dino_r50_4scale_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/dino/dino_r50_4scale_1x_coco_train_infer_python.txt @@ -50,7 +50,7 @@ inference:./deploy/python/infer.py --run_benchmark:False --trt_max_shape:1600 ===========================train_benchmark_params========================== -batch_size:4 +batch_size:2 fp_items:fp32|fp16 epoch:1 repeat:2