diff --git a/test_tipc/configs/PPLCNet/PPLCNet_x2_5_train_infer_python.txt b/test_tipc/configs/PPLCNet/PPLCNet_x2_5_train_infer_python.txt index 717e67e56..5b5951b55 100644 --- 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