Add matmul and elementwise into amp_custom_black_list for ch_PP-OCRv2_rec_distillation (#8360)

931edad1 · niuliling123 · GitHub · cf2e9eb7 · 931edad1
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

configs/rec/ch_PP-OCRv2/ch_PP-OCRv2_rec_distillation.yml configs/rec/ch_PP-OCRv2/ch_PP-OCRv2_rec_distillation.yml +1 -0

未找到文件。
--- a/configs/rec/ch_PP-OCRv2/ch_PP-OCRv2_rec_distillation.yml
+++ b/configs/rec/ch_PP-OCRv2/ch_PP-OCRv2_rec_distillation.yml
@@ -19,6 +19,7 @@ Global:
  use_space_char: true
  distributed: true
  save_res_path: ./output/rec/predicts_pp-OCRv2_distillation.txt
+  amp_custom_black_list: ['matmul','matmul_v2','elementwise_add']
 Optimizer: