nodefluxio / vortex

A Deep Learning Model Development Framework for Computer Vision
27 stars 6 forks source link

Amp trainer #63

Closed alphinside closed 3 years ago

alphinside commented 4 years ago

Type of changes

Please check the type of change your PR introduces: - [ ] Bugfix - [x] Feature - [ ] Code style update (formatting, renaming) - [ ] Refactoring (no functional changes, no api changes) - [ ] Build related changes - [x] Documentation content changes - [ ] Other (please describe): ## What is the current behavior? We don't have any mixed precision training - close #62 Related issues : - depends on #17 , Need to address issue for pytorch version higher than 1.4.0 ## What is the new behavior?

Checklist

triwahyuu commented 3 years ago

we'll close this PR because it is no longer served purpose in pytorch lightning trainer backend @alphinside