fix det_v3 (#10180)

a288a3aa · Zhang Ting · GitHub · 2d44a71b · a288a3aa
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml +2 -1

未找到文件。
--- 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