From 931edad131a4991c28555088df9f41635f112973 Mon Sep 17 00:00:00 2001
From: niuliling123 <51102941+niuliling123@users.noreply.github.com>
Date: Fri, 18 Nov 2022 14:34:13 +0800
Subject: [PATCH] Add matmul and elementwise into amp_custom_black_list for
 ch_PP-OCRv2_rec_distillation (#8360)

---
 configs/rec/ch_PP-OCRv2/ch_PP-OCRv2_rec_distillation.yml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/configs/rec/ch_PP-OCRv2/ch_PP-OCRv2_rec_distillation.yml b/configs/rec/ch_PP-OCRv2/ch_PP-OCRv2_rec_distillation.yml
index ab48b997..9bddcb29 100644
--- 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:
-- 
GitLab