pytorch-blockswap icon indicating copy to clipboard operation
pytorch-blockswap copied to clipboard

Published 20 hours ago • verified publisher icon BayesWatch

Any plan to release the configurations that can reproduce the ImageNet/COCO results?

Open lcmeng opened this issue 4 years ago • 3 comments

It'll be great to try out something other than CIFAR10. TIA.

lcmeng avatar Dec 30 '20 05:12 lcmeng

I've added them now (under genotypes/bs_imagenet_N.csv), as well as a ResNet model definition to models/resnet.py.

jack-willturner avatar Dec 30 '20 08:12 jack-willturner

I'll need to do some testing and then I'll close this issue:

  • [ ] double check the genotypes are correct
  • [ ] add an imagenet/coco training script
  • [ ] include a seed for reproducing the results
  • [ ] add the baselines (downscaled resnets)

jack-willturner avatar Dec 31 '20 11:12 jack-willturner

Thank you. I'll give it a spin as soon as you think it's in good shape. BTW, the README uses--param_goal which is no longer a valid argument for model_generator.py.

lcmeng avatar Jan 01 '21 05:01 lcmeng