pyssl

pyssl copied to clipboard

Hello, Congratulation for this great work. I have a question regarding the loss function from SWAV. ``` def cross_entropy_loss(q, p): return torch.mean(torch.sum(q * F.log_softmax(p, dim=1), dim=1)) ``` That ``F.log_softmax`` will...

wallaceloos

SimCLRv2 memory bank

1

I looked at the SimCRLv2 implementation and I don't see the memory bank that you implement in the version given in the paper, which is also explained in your post...

sfandres94

Pyssl on 3D data (images)

4

Great package. I am wondering if there is possibility to use this package for 3D volumetric data (images)?

NimaMojtahedi