diff --git a/test_tipc/config/ResNet/ResNet50_train_ampfp16_infer_python.txt b/test_tipc/config/ResNet/ResNet50_train_ampfp16_infer_python.txt index 8256c2a3957ba954b1aee2fdef0bb5f6c4711760..3c7592779c5f1a15279eaa1cb7bdd564a231904e 100644 --- a/test_tipc/config/ResNet/ResNet50_train_ampfp16_infer_python.txt +++ b/test_tipc/config/ResNet/ResNet50_train_ampfp16_infer_python.txt @@ -51,6 +51,6 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml null:null null:null ===========================train_benchmark_params========================== -batch_size:128 -fp_items:amp_fp16 +batch_size:128|256 +fp_items:ampfp16 epoch:1 diff --git a/test_tipc/config/ResNet/ResNet50_train_purefp16_infer_python.txt b/test_tipc/config/ResNet/ResNet50_train_purefp16_infer_python.txt index b48e5e1c4fea914e63387a6c2b9665026a8b5010..907d579ecc01740922e3c5e06b5087063685224e 100644 --- a/test_tipc/config/ResNet/ResNet50_train_purefp16_infer_python.txt +++ b/test_tipc/config/ResNet/ResNet50_train_purefp16_infer_python.txt @@ -51,6 +51,6 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml null:null null:null ===========================train_benchmark_params========================== -batch_size:128 -fp_items:pure_fp16 +batch_size:128|256 +fp_items:purefp16 epoch:1