diff --git a/test_tipc/configs/picodet/picodet_lcnet_1_0x_416_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_lcnet_1_0x_416_coco_train_infer_python.txt index f79e641526e1973653cbde1a366060fa3f0e2886..a1019e3a7ca1cbad6bdb644b0f653c36365b7584 100644 --- a/test_tipc/configs/picodet/picodet_lcnet_1_0x_416_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_lcnet_1_0x_416_coco_train_infer_python.txt @@ -51,10 +51,10 @@ inference:./deploy/python/infer.py null:null ===========================train_benchmark_params========================== batch_size:90 -fp_items:fp32|fp16 +fp_items:fp32|amp epoch:1 repeat:1 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile flags:null ===========================infer_benchmark_params=========================== -numpy_infer_input:3x416x416_2.npy \ No newline at end of file +numpy_infer_input:3x416x416_2.npy diff --git a/test_tipc/configs/picodet/picodet_s_320_coco_lcnet_train_infer_python.txt b/test_tipc/configs/picodet/picodet_s_320_coco_lcnet_train_infer_python.txt index ddbf6f64f685fa83fb30e1dcaa312bce07a68b48..e5832f1b6e18823d426dc95e46a0879bdbda5c2d 100644 --- a/test_tipc/configs/picodet/picodet_s_320_coco_lcnet_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_s_320_coco_lcnet_train_infer_python.txt @@ -51,7 +51,7 @@ inference:./deploy/python/infer.py null:null ===========================train_benchmark_params========================== batch_size:64 -fp_items:fp32|amp +fp_items:fp32|fp16 epoch:1 repeat:1 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile diff --git a/test_tipc/configs/picodet/picodet_s_320_coco_train_infer_python.txt b/test_tipc/configs/picodet/picodet_s_320_coco_train_infer_python.txt index 01aa4644ca5269d59456f30d3dad656b30000a00..94f2ca1b09c2472442a2a1fff7c4135ee259009a 100644 --- a/test_tipc/configs/picodet/picodet_s_320_coco_train_infer_python.txt +++ b/test_tipc/configs/picodet/picodet_s_320_coco_train_infer_python.txt @@ -51,10 +51,10 @@ inference:./deploy/python/infer.py null:null ===========================train_benchmark_params========================== batch_size:64 -fp_items:fp32|fp16 +fp_items:fp32|amp epoch:1 repeat:50 --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile flags:null ===========================infer_benchmark_params=========================== -numpy_infer_input:3x320x320_2.npy \ No newline at end of file +numpy_infer_input:3x320x320_2.npy