diff --git a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml index d0c1de28ced5cca64f13fce75dbe7e1311e3d20d..252d1599776a893b882723a9f7329c99458f3dc4 100644 --- a/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml +++ b/configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml @@ -18,6 +18,7 @@ Global: save_res_path: ./checkpoints/det_db/predicts_db.txt distributed: true d2s_train_image_shape: [3, -1, -1] + amp_dtype: bfloat16 Architecture: name: DistillationModel @@ -222,4 +223,4 @@ Eval: shuffle: False drop_last: False batch_size_per_card: 1 # must be 1 - num_workers: 2 \ No newline at end of file + num_workers: 2