From 5827b2c9d18efcc62a83e77e634632fb440e0cf8 Mon Sep 17 00:00:00 2001 From: dongshuilong Date: Thu, 16 Jun 2022 03:12:00 +0000 Subject: [PATCH] add r50 dynamic fp16 train for benchmark --- .../config/ResNet/ResNet50_train_ampfp16_infer_python.txt | 4 ++-- .../config/ResNet/ResNet50_train_purefp16_infer_python.txt | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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 8256c2a3..3c759277 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 b48e5e1c..907d579e 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 -- GitLab