Welcome to NFNets PyTorch’s documentation!

NFNets-PyTorch is an implementation of the paper: “High-Performance Large-Scale Image Recognition Without Normalization “. Original paper can be found at arxiv. You can find other implementations at PapersWithCode.

Install

Stable release

pip3 install nfnets-pytorch

Latest code

pip3 install git+https://github.com/vballoli/nfnets-pytorch

Sample usage

import torch
from torch import nn
from torchvision.models import resnet18

from nfnets import replace_conv, SGD_AGC

model = resnet18()
replace_conv(model)
optim = SGD_AGC(model.parameters(), 1e-3)