train loss = nan #1
Description
When I try to train with AMCLSimCLR on CIFAR10 dataset, the loss value is always nan, moreover under src/models.AMCLSimCLRModle, in 'self.criterion = NTXentLoss()' NTXentLoss is not defined, is it also 'lightly.loss.NTXentLoss()'?