Add matmul_v2 and elementwise_add to black list for amp o1 and o2 (#8734)

2026-09-24 23:33:08 +08:00 · 2022-12-29 13:41:58 +08:00
parent d6ef2ac4f2
commit bdfa1d2fa8
1 changed files with 1 additions and 0 deletions
@@ -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