/pytorch-lamb

PyTorch implementation of LAMB for ImageNet/ResNet-50 training

Primary LanguagePythonOtherNOASSERTION