Adding 'ctc_loss' to the list of FP32 funcs (#722)

93cabd5d · Saransh Karira · GitHub · 1733946a · 93cabd5d
Unverified Commit 93cabd5d authored Feb 25, 2020 by Saransh Karira Committed by GitHub Feb 24, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

apex/amp/lists/functional_overrides.py apex/amp/lists/functional_overrides.py +2 -1

No files found.
--- a/apex/amp/lists/functional_overrides.py
+++ b/apex/amp/lists/functional_overrides.py
@@ -63,7 +63,8 @@ FP32_FUNCS = [
    'binary_cross_entropy_with_logits',
    'smooth_l1_loss',
    'soft_margin_loss',
-    'triplet_margin_loss'
+    'triplet_margin_loss',
+    'ctc_loss'
 ]

 BANNED_FUNCS = [