search

n2cholas / jax-resnet

Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
https://pypi.org/project/jax-resnet/
MIT License
103 stars 8 forks source link

WideResNet-28-10 #8

Open nalzok opened 2 years ago

nalzok commented 2 years ago

Hi there, this repo is very useful, but could you add WideResNet-28-10? It is highlighted in the official repo and is the best-performing model on CIFAR-10 in RobustBench, so I believe many people will find it helpful.

n2cholas commented 2 years ago

Will look into it, thanks for bring it to my attention! Would also be open to PRs if you are able.

nalzok commented 2 years ago

I did some research and opened an issue on Flax. I would love to contribute, but the problem is I don't really understand how WideResNet-28-10 is defined.