From 483c0383e5eac6ae45a0aa25279ee4d562777cc8 Mon Sep 17 00:00:00 2001 From: root Date: Fri, 29 Jul 2022 08:53:05 +0000 Subject: [PATCH] tipc: set bs=256 to avoid oom --- .../configs/PPLCNet/PPLCNet_x2_5_train_infer_python.txt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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 37368f74..5db7cb4e 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 @@ -20,7 +20,7 @@ distill_train:null null:null null:null ## -===========================eval_params=========================== +===========================eval_params=========================== eval:tools/eval.py -c ppcls/configs/ImageNet/PPLCNet/PPLCNet_x2_5.yaml null:null ## @@ -50,10 +50,10 @@ 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 flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 ===========================infer_benchmark_params========================== -random_infer_input:[{float32,[3,224,224]}] \ No newline at end of file +random_infer_input:[{float32,[3,224,224]}] -- GitLab