tipc: set bs=256 to avoid oom

483c0383 · root · Tingquan Gao · b9de09e5 · 483c0383
显示空白变更内容
内联并排

Showing with 3 addition and 3 deletion

test_tipc/configs/PPLCNet/PPLCNet_x2_5_train_infer_python.txt ..._tipc/configs/PPLCNet/PPLCNet_x2_5_train_infer_python.txt +3 -3

未找到文件。
--- 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
 fp_items:fp32|fp16
 epoch:1
 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile