TTN-YKK / Clustering_friendly_representation_learning

Other
53 stars 9 forks source link

How to Cluster on a Actual(Custom) Dataset? #5

Closed avernal1819 closed 8 months ago

avernal1819 commented 8 months ago

Firstly, I would like to express my deep gratitude for your IDFD.

I have saved the checkpoint file using the code

And, my data size is 512, 21 classes

torch.save({'optimizer': optimizer.state_dict(), 'model': net.state_dict(), 'npc': npc.state_dict(), 'epoch': epoch + 1}, f'{args.pretrained}/checkpoint.pth'). 

Now, I am trying to perform clustering on my custom dataset, but I am facing difficulties with the code configuration.

Could you possibly offer some assistance?