Make lobes use fp32 when AMP is active - #2295

Merged

TParcollet merged 11 commits into

speechbrain:developfrom

asumagic:fp32-lobes

Dec 8, 2023

Commits on Dec 8, 2023

Added utils.autocast with a fwd_default_precision function
asumagic
committed
cb848c6
Decorate all lobes to require float32 precision in AMP
asumagic
committed
8548c3f
Fix trailing space in docstring
asumagic
committed
022b5a6
Less confusing doc for fwd_default_precision
asumagic
committed
e602e9e
Be explicit that only fp inputs are affected by fwd_default_precision
asumagic
committed
5288041
Typo in docstring
asumagic
committed
b1b71e5
Remove dtype annotation that is broken for some reason
asumagic
committed
68c03bf
Precommit checks will be the end of me
asumagic
committed
89aedc9
Fix tests
asumagic
committed
f8b90f7
Add docstring to precision wrapper function
asumagic
committed
da8afa7
Fix style check again..
asumagic
committed
f8b9cba