/LARS-ImageNet-PyTorch

Accuracy 77%. Large batch deep learning optimizer LARS for ImageNet with PyTorch and ResNet, using Horovod for distribution. Optional accumulated gradient and NVIDIA DALI dataloader.

Primary LanguagePythonMIT LicenseMIT

Stargazers