fix batch_size to avoid oom

test_tipc/configs/DeiT/DeiT_base_patch16_224_train_infer_python.txt

@@ -20,7 +20,7 @@ distill_train:null
 to_static_train:-o Global.to_static=True
 null:null
 ##
-===========================eval_params=========================== 
+===========================eval_params===========================
 eval:tools/eval.py -c ppcls/configs/ImageNet/DeiT/DeiT_base_patch16_224.yaml
 null:null
 ##
@@ -51,7 +51,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 null:null
 null:null
 ===========================train_benchmark_params==========================
-batch_size:256
+batch_size:128
 fp_items:fp32|fp16
 epoch:1
 model_type:norm_train|to_static_train

test_tipc/configs/DeiT/DeiT_tiny_patch16_224_train_infer_python.txt

@@ -20,7 +20,7 @@ distill_train:null
 to_static_train:-o Global.to_static=True
 null:null
 ##
-===========================eval_params=========================== 
+===========================eval_params===========================
 eval:tools/eval.py -c ppcls/configs/ImageNet/DeiT/DeiT_tiny_patch16_224.yaml
 null:null
 ##
@@ -51,7 +51,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 null:null
 null:null
 ===========================train_benchmark_params==========================
-batch_size:256
+batch_size:128
 fp_items:fp32|fp16
 epoch:1
 model_type:norm_train|to_static_train

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:256|512
+batch_size:256
 fp_items:fp32|fp16
 epoch:1
 model_type:norm_train|to_static_train