vballoli / nfnets-pytorch

NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/

Home Page:https://nfnets-pytorch.readthedocs.io/en/latest/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Thanks a lot for you work . Could you release pretrained weight ?

wtjgh opened this issue · comments

commented

Unfortunately, I don't have any personal compute to release any pre-trained weights. But I'll be happy to host it as a release through GitHub if anyone reading this could possibly use their GPUs and can make a PR.