/pytorch-lars

PyTorch implementation of LARS (Layer-wise Adaptive Rate Scaling)

Primary LanguagePythonMIT LicenseMIT

Watchers