Pytorch 1.6, mixed precision update

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%.