byol-pytorch
byol-pytorch copied to clipboard
How to cluster/predict images?
Hi,
I have trained using examples given with pytorch-lightning
. I couldn't find code to do clustering of images after training. How can I find which image falls in which cluster? Is there any predictor API?
I want to do something like this
@laxmimerit anything you found out?
Thank you