ai-research-code icon indicating copy to clipboard operation
ai-research-code copied to clipboard

[Mixed Precision DNNs]: ImageNet codebase?

Open UtkarshSaxena1 opened this issue 3 years ago • 1 comments

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

UtkarshSaxena1 avatar Feb 01 '22 14:02 UtkarshSaxena1

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

TE-LukasMauch avatar Feb 04 '22 09:02 TE-LukasMauch