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

bdfa1d2f · niuliling123 · GitHub · d6ef2ac4 · bdfa1d2f
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

configs/table/table_mv3.yml configs/table/table_mv3.yml +1 -0

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