-
Notifications
You must be signed in to change notification settings - Fork 38
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Can you guide me for creating self-supervised .ckpt file for custom dataset? #19
Comments
Hi @saifhassan-phd , Our self-supervised model is adapted from our self-supervised learning work CACR. Please feel free to check this repo, where you can find how to use CACR to learn the self-supervised model on your dataset: https://github.com/JegZheng/CACR-SSL |
ok, Thanks for your quick response. |
@JegZheng @XzwHan , I have gone through https://github.com/JegZheng/CACR-SSL repo and completed cifar10 experiment and it generated encoder.pth file using Thanks |
Specifically, can you guide, how you generate |
We have provided the proxy function to load .pth file. Please check our code here (CARD/classification/card_classification.py, L208): https://github.com/XzwHan/CARD/blob/ebe2f2ae95dc7a7a95e6a71c0c8e1cabf8451087/classification/card_classification.py#L208C4-L208C5 Just put the .pth file in the corresponding folder and rename it as aux_ckpt.pth and it shall be ready.
|
I didn't understand, Let me rephrase, I want to know how you produced |
I mean how can I generate these |
I have tried that proxy function, it causes keyerror on line 210, tried |
The .pth file saved from CACR repo is directly the state_dict, and the .ckpt in CARD is a dictionary, where the state_dict is stored under the key 'state_dict'. This is why:
Let's do a simpler solution without changing the code. Could you please try:
For example, |
I got it, I was doing mistake by using different backbone while SSL and then in CARD, got it. Error Solved. Thanks for your quick response. |
Thank you @XzwHan @JegZheng for this wonderful contribution, I have implemented this and want to use this model for my research project. So can you guide me for creating self-supervised .ckpt file for custom dataset (Any image classification dataset)?
I have gone through paper and code but unable to understand how you have developed .ckpt file for self-supervised model.
Thank you in advance.
The text was updated successfully, but these errors were encountered: