Skip to content

GitLab

Unverified Commit 93cabd5d authored by Saransh Karira's avatar Saransh Karira Committed by GitHub
Browse files

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

parent 1733946a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
apex/amp/lists/functional_overrides.py
View file @ 93cabd5d
...@@ -63,7 +63,8 @@ FP32_FUNCS = [ ...@@ -63,7 +63,8 @@ FP32_FUNCS = [
'binary_cross_entropy_with_logits', 'binary_cross_entropy_with_logits',
'smooth_l1_loss', 'smooth_l1_loss',
'soft_margin_loss', 'soft_margin_loss',
'triplet_margin_loss' 'triplet_margin_loss',
'ctc_loss'
] ]
BANNED_FUNCS = [ BANNED_FUNCS = [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment