loss nan using custom dataset

wangz10 / contrastive_loss

Experiments with supervised contrastive learning methods with different loss functions

213 stars 35 forks source link

loss nan using custom dataset #13

Open ramdhan1989 opened 3 years ago

ramdhan1989 commented 3 years ago

Hi, it is strange when using custom dataset the loss is nan (I just tried supervised_nt_xent_loss and max_margin_contrastive_loss). do you have any idea ?

thanks