NUS-HPC-AI-Lab / 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.
38Updated 3 years ago

Related projects

Alternatives and complementary repositories for LARS-ImageNet-PyTorch