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¶
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)