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

Loading DenseNet Checkpoints for CIFAR-10/100 #9

Open
superctj opened this issue Sep 21, 2021 · 2 comments
Open

Loading DenseNet Checkpoints for CIFAR-10/100 #9

superctj opened this issue Sep 21, 2021 · 2 comments

Comments

@superctj
Copy link

Thank you for open sourcing this great project. The way you saved the Cifar10/100 checkpoints makes it hard to reuse the pre-trained Densenets in another project. Could you please release the checkpoints that can be loaded like

model = models.DenseNet3(100, int(args.num_classes))

@superctj
Copy link
Author

Never mind. The workaround is to keep the same directory structure and load the entire model

@saandeepa93
Copy link

Can you please let me know the solution for this?

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

2 participants