giakoumoglou / pyssl Goto Github PK

I have one question about the code, I think in main.py line 49, it should be model.forward(image). can you give me some explanation? many thanks.

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 on Medium. Is that correct? I wanted to use your library but I need the full implementation of that method.

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

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 return a negative number. Should it be -F.log_softmax(p, dim=1)?

Thank you.

Hi, thanks for the open source. I want to know the performance of these self-supervised methods in this framework. Can you show me? thanks again.