diff --git a/test_tipc/configs/StyleGANv2/train_infer_python.txt b/test_tipc/configs/StyleGANv2/train_infer_python.txt index 423773bffc144530775a739806eee5242f95b888..780ea2564dbe1abcbd4bcd5322fe2d1db5250c41 100644 --- a/test_tipc/configs/StyleGANv2/train_infer_python.txt +++ b/test_tipc/configs/StyleGANv2/train_infer_python.txt @@ -50,10 +50,10 @@ null:null --benchmark:True null:null ===========================train_benchmark_params========================== -batch_size:8|16 +batch_size:8 fp_items:fp32 epoch:100 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile flags:null ===========================infer_benchmark_params========================== -random_infer_input:[{float32,[1,512]}, {float32,[1]}] \ No newline at end of file +random_infer_input:[{float32,[1,512]}, {float32,[1]}]