search

lucidrains / byol-pytorch

Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch
MIT License
1.76k stars 248 forks source link

grayscale images #69

Open matteoguarrera opened 3 years ago

matteoguarrera commented 3 years ago

I would love to have the possibility to use the library directly with grayscale images, without converting them again to RGB