Nfnets-pytorch

Latest version: v0.1.3

Safety actively analyzes 706267 Python packages for vulnerabilities to keep your Python projects secure.

Scan your dependencies

Page 2 of 3

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

Fixes the step method of AGC

1. Incopororated bfialkoff's fix https://github.com/vballoli/nfnets-pytorch/issues/3.
2. Add a generic AGC implementation.

Fixes of replace_conv and replaces `BatchNorm2d` with `Identity`.

Has known vulnerabilities

Previous Next