casm1

Results 2 comments of casm1

请问是直接使用vq计算InfoNCE吗?InfoNCE是这样计算吗: lsoftmax = nn.LogSoftmax(dim=0) total1 = torch.mm(a_vq, torch.transpose(b_vq, 0, 1)) nce += torch.sum(torch.diag(self.lsoftmax(total1)))

好的,感谢!