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

[Mixed Precision DNNs]: ImageNet codebase? #47

Open
UtkarshSaxena1 opened this issue Feb 1, 2022 · 1 comment
Open

[Mixed Precision DNNs]: ImageNet codebase? #47

UtkarshSaxena1 opened this issue Feb 1, 2022 · 1 comment

Comments

@UtkarshSaxena1
Copy link

Can you share the working codebase to replicate paper results for ImageNet with MixedPrecisionQuantization?

@TE-LukasMauch
Copy link

Thank's for the interest in our work. Following your request, we will start the process of releasing the training code for the ImageNet experiments as well. However, this might take a while.

In the mean time, I can share the configuration files for the training with you. You should be able to adapt the existing code, provided that you have a working dataloader for ImageNet. The naming convention for the experiments is similar to the CIFAR10 experiments.

I will announce here, as soon as the code has been released.
train_mobilenet2.txt
train_resnet18.txt

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