Skip to content

GitLab

Commit 47e9cae5 authored by Marek Kolodziej's avatar Marek Kolodziej
Browse files

Run interpolation in fp32 because it's faster

parent 3f87614f
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
apex/amp/lists/functional_overrides.py
View file @ 47e9cae5
...@@ -27,6 +27,10 @@ FP16_FUNCS = [ ...@@ -27,6 +27,10 @@ FP16_FUNCS = [
] ]
FP32_FUNCS = [ FP32_FUNCS = [
# Interpolation/Upsampling
'interpolate',
# Pointwise # Pointwise
'softplus', 'softplus',
'softmin', 'softmin',
......
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