From bdfa1d2fa815e5c365df6861e712fe373e030aa6 Mon Sep 17 00:00:00 2001
From: niuliling123 <51102941+niuliling123@users.noreply.github.com>
Date: Thu, 29 Dec 2022 13:41:58 +0800
Subject: [PATCH] Add matmul_v2 and elementwise_add to black list for amp o1
 and o2 (#8734)

---
 configs/table/table_mv3.yml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/configs/table/table_mv3.yml b/configs/table/table_mv3.yml
index 9d286f41..50c84393 100755
--- a/configs/table/table_mv3.yml
+++ b/configs/table/table_mv3.yml
@@ -20,6 +20,7 @@ Global:
   max_text_length: &max_text_length 500
   box_format: &box_format 'xyxy' # 'xywh', 'xyxy', 'xyxyxyxy'
   infer_mode: False
+  amp_custom_black_list: ['matmul_v2','elementwise_add']
 
 Optimizer:
   name: Adam
-- 
GitLab