Supervised-Contrastive-Learning-in-TensorFlow-2
Supervised-Contrastive-Learning-in-TensorFlow-2 copied to clipboard
Published
20 hours ago •
sayakpaul
sayakpaul
Could you provide the trained weights please
Hi I want first thank you for your good work. I have been modifying the notebook to run on my own dataset. Size is close to 1000 samples, 5 classes.
I trained the encoder and projector using your notebook, Resnet50 with SGD optimizer, but I haven't been able to produce the TSNA and PCA graphs as you did on the notebook visualizing ImageNet subset.
Even I switched back to using the ImageNet subset, the visuallization result still looks nothing like the results you produced in the original notebook.
If you would be so kind, could you provide the trained weight files that you used to produce those visualization results?
Thank you.
