Are all the weights available from 16-bit precision? Not 16-mixed?

vturrisi / solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning

MIT License

1.38k stars 181 forks source link

Are all the weights available from 16-bit precision? Not 16-mixed? #367

Closed LightingMc closed 10 months ago

LightingMc commented 10 months ago

User warning: 16 is supported for historical reasons, but its usage is discouraged. Please set your precision to 16-mixed instead!

vturrisi commented 10 months ago

Most models and scripts are pretty old, so they were executed when 16-mixed was not supported in lightning. I'll eventually update all scripts and the imports for lightning, but the models will be kept at the current precision.