├── .gitattributes ├── .gitignore ├── LICENSE ├── README.md ├── dataset.py ├── default_config.yaml ├── demo.ipynb ├── eval.py ├── nfnets ├── __init__.py ├── model.py ├── optim.py └── pretrained.py ├── pretrained └── README.md ├── pyproject.toml ├── requirements.txt ├── setup.cfg └── train.py /.gitattributes: -------------------------------------------------------------------------------- 1 | *.ipynb linguist-documentation -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/README.md -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/dataset.py -------------------------------------------------------------------------------- /default_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/default_config.yaml -------------------------------------------------------------------------------- /demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/demo.ipynb -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/eval.py -------------------------------------------------------------------------------- /nfnets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/nfnets/__init__.py -------------------------------------------------------------------------------- /nfnets/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/nfnets/model.py -------------------------------------------------------------------------------- /nfnets/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/nfnets/optim.py -------------------------------------------------------------------------------- /nfnets/pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/nfnets/pretrained.py -------------------------------------------------------------------------------- /pretrained/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/pretrained/README.md -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/setup.cfg -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/benjs/nfnets_pytorch/HEAD/train.py --------------------------------------------------------------------------------