Nfnets Pytorch Versions Save

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

0.1.3

3 years ago

0.1.2

3 years ago

0.1.1

3 years ago

0.1.0

3 years ago

0.0.9

3 years ago

0.0.8

3 years ago

0.0.7

3 years ago

0.0.6

3 years ago
  1. Adds the AGC to the main imports
  2. Add ignore_agc to the AGC arguments to align with the papers' recommendation.

0.0.5

3 years ago

0.0.4

3 years ago

Fixes the step method of AGC