diff --git a/test_tipc/benchmark_train.sh b/test_tipc/benchmark_train.sh index d058b1daf21348cd88735ef63536f5605746599e..b80e7257ce924b357650c64910a1e323389680d2 100644 --- a/test_tipc/benchmark_train.sh +++ b/test_tipc/benchmark_train.sh @@ -69,8 +69,8 @@ PARAMS=$3 IFS=$'\n' # parser params from train_benchmark.txt sed -i 's/ -o DataLoader.Train.sampler.shuffle=False//g' $FILENAME -sed -i 's/ -o DataLoader.Train.loader.num_workers=0/-o Global.print_batch_step=1/g' $FILENAME -sed -i 's/-o DataLoader.Train.loader.use_shared_memory=False/-o Global.eval_during_train=False/g' $FILENAME +sed -i 's/ -o DataLoader.Train.loader.num_workers=0/ -o Global.print_batch_step=1/g' $FILENAME +sed -i 's/-o DataLoader.Train.loader.use_shared_memory=False/ -o Global.eval_during_train=False/g' $FILENAME dataline=`cat $FILENAME` # parser params IFS=$'\n' diff --git a/test_tipc/config/Twins/alt_gvt_base_train_infer_python.txt b/test_tipc/config/Twins/alt_gvt_base_train_infer_python.txt index 6dde91b4ac0d6a522e322a2efc686d38122f99ce..ef4b566c4627de9f9aee6fe7412ca6646e77ef7a 100644 --- a/test_tipc/config/Twins/alt_gvt_base_train_infer_python.txt +++ b/test_tipc/config/Twins/alt_gvt_base_train_infer_python.txt @@ -51,10 +51,10 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml null:null null:null ===========================train_benchmark_params========================== -batch_size:64|176 +batch_size:64|144 fp_items:fp32 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]}]