Skip to content
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

Model mismatch occurs when training student network with own data set #88

Open
rogressing opened this issue Oct 31, 2023 · 2 comments
Open

Comments

@rogressing
Copy link

I used my own data set for training. According to the training steps given by the author, I first pre-trained the supernet and parameters, and then obtained the search file to train the teacher network. **After training the teacher network, I distilled it to the student network again, but when the student network is loaded, The problem occurs as follows**:

image
And the prompt is that the loaded model and the current model do not match the prompt in the code here:
image
I would like to ask the author or other experts if they have encountered this kind of problem and how to solve it? And what I am very confused about is that the accuracy of my teacher's network training and verification is very low, only about 29%.

@rogressing
Copy link
Author

I used my own data set for training. According to the training steps given by the author, I first pre-trained the supernet and parameters, and then obtained the search file to train the teacher network. After training the teacher network, I distilled it to the student network again, but when the student network is loaded, The problem occurs as follows:

@rogressing
Copy link
Author

Thank you ,It's been solved

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant