modify

2023-03-16 08:21:59 +00:00 · 2023-03-16 08:21:59 +00:00 · 46fe37597b
parent 174ae53722
commit 46fe37597b
1 changed files with 1 additions and 1 deletions
--- a/test_tipc/configs/PPLCNet/PPLCNet_x2_5_train_infer_python.txt
+++ b/test_tipc/configs/PPLCNet/PPLCNet_x2_5_train_infer_python.txt
@ -50,7 +50,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.benchmark:False
 null:null
 ===========================train_benchmark_params==========================
-batch_size:512
+batch_size:256|512
 fp_items:fp32|fp16
 epoch:1
 model_type:norm_train|to_static_train