Pytorch 1.6, mixed precision update
Pre-release
Pre-release
In v0.1.1:
- The pytorch version has been updated to 1.6 (also in the Dockerfile) (and pytorch 1.6 is now required)
- Mixed precision support with the
--mixed-precision
flag. If you have supporting hardware this can speed up training with more than 33% and reduce memory by more than 40%.