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 717e67e5656c9f40de681f3e2c19817dc17e3c48..5b5951b55b7079e02d4e2eb8697eead80d7eeb5b 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