├── .gitignore ├── LICENSE ├── README.md ├── colab └── start_on_colab.ipynb ├── hubconf.py ├── pytorch_cifar_models ├── __init__.py ├── mobilenetv2.py ├── repvgg.py ├── resnet.py ├── shufflenetv2.py ├── vgg.py └── vit.py ├── setup.py └── tests └── pytorch_cifar_models ├── test_mobilenetv2.py ├── test_repvgg.py ├── test_resnet.py ├── test_shufflenetv2.py ├── test_vgg.py └── test_vit.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/README.md -------------------------------------------------------------------------------- /colab/start_on_colab.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/colab/start_on_colab.ipynb -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/hubconf.py -------------------------------------------------------------------------------- /pytorch_cifar_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/pytorch_cifar_models/__init__.py -------------------------------------------------------------------------------- /pytorch_cifar_models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/pytorch_cifar_models/mobilenetv2.py -------------------------------------------------------------------------------- /pytorch_cifar_models/repvgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/pytorch_cifar_models/repvgg.py -------------------------------------------------------------------------------- /pytorch_cifar_models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/pytorch_cifar_models/resnet.py -------------------------------------------------------------------------------- /pytorch_cifar_models/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/pytorch_cifar_models/shufflenetv2.py -------------------------------------------------------------------------------- /pytorch_cifar_models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/pytorch_cifar_models/vgg.py -------------------------------------------------------------------------------- /pytorch_cifar_models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/pytorch_cifar_models/vit.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/setup.py -------------------------------------------------------------------------------- /tests/pytorch_cifar_models/test_mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/tests/pytorch_cifar_models/test_mobilenetv2.py -------------------------------------------------------------------------------- /tests/pytorch_cifar_models/test_repvgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/tests/pytorch_cifar_models/test_repvgg.py -------------------------------------------------------------------------------- /tests/pytorch_cifar_models/test_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/tests/pytorch_cifar_models/test_resnet.py -------------------------------------------------------------------------------- /tests/pytorch_cifar_models/test_shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/tests/pytorch_cifar_models/test_shufflenetv2.py -------------------------------------------------------------------------------- /tests/pytorch_cifar_models/test_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/tests/pytorch_cifar_models/test_vgg.py -------------------------------------------------------------------------------- /tests/pytorch_cifar_models/test_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenyaofo/pytorch-cifar-models/HEAD/tests/pytorch_cifar_models/test_vit.py --------------------------------------------------------------------------------