[FEATURE] Add support for Automatic Mixed Precision Training

nodefluxio / vortex

A Deep Learning Model Development Framework for Computer Vision

27 stars 6 forks source link

Closed alphinside closed 3 years ago

alphinside commented 4 years ago

Is your feature request related to a problem? Please describe. Nope

Describe the solution you'd like Add a new trainer driver for automatic mixed precision training

Additional context This issue is dependent on issue #17 because AMP is available from Pytorch 1.6.0