Closed avernal1819 closed 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?
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
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?