├── .gitignore ├── README.md ├── main.py ├── main_amp.py ├── main_graph.py ├── main_qat.py ├── models ├── __init__.py ├── alexnet.py ├── densenet.py ├── dla.py ├── dla_simple.py ├── dpn.py ├── efficientnet.py ├── googlenet.py ├── lenet.py ├── mobilenet.py ├── mobilenetv2.py ├── pnasnet.py ├── preact_resnet.py ├── regnet.py ├── resnet.py ├── resnext.py ├── senet.py ├── shufflenet.py ├── shufflenetv2.py └── vgg.py ├── resnet18_cifar10_loss_fp16.txt ├── resnet18_cifar10_loss_fp32.txt └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | log/ 2 | data/ 3 | __pycache__/ 4 | checkpoint/ 5 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/README.md -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/main.py -------------------------------------------------------------------------------- /main_amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/main_amp.py -------------------------------------------------------------------------------- /main_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/main_graph.py -------------------------------------------------------------------------------- /main_qat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/main_qat.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/alexnet.py -------------------------------------------------------------------------------- /models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/densenet.py -------------------------------------------------------------------------------- /models/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/dla.py -------------------------------------------------------------------------------- /models/dla_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/dla_simple.py -------------------------------------------------------------------------------- /models/dpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/dpn.py -------------------------------------------------------------------------------- /models/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/efficientnet.py -------------------------------------------------------------------------------- /models/googlenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/googlenet.py -------------------------------------------------------------------------------- /models/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/lenet.py -------------------------------------------------------------------------------- /models/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/mobilenet.py -------------------------------------------------------------------------------- /models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/mobilenetv2.py -------------------------------------------------------------------------------- /models/pnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/pnasnet.py -------------------------------------------------------------------------------- /models/preact_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/preact_resnet.py -------------------------------------------------------------------------------- /models/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/regnet.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/resnext.py -------------------------------------------------------------------------------- /models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/senet.py -------------------------------------------------------------------------------- /models/shufflenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/shufflenet.py -------------------------------------------------------------------------------- /models/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/shufflenetv2.py -------------------------------------------------------------------------------- /models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/models/vgg.py -------------------------------------------------------------------------------- /resnet18_cifar10_loss_fp16.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/resnet18_cifar10_loss_fp16.txt -------------------------------------------------------------------------------- /resnet18_cifar10_loss_fp32.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/resnet18_cifar10_loss_fp32.txt -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BBuf/oneflow-cifar/HEAD/utils.py --------------------------------------------------------------------------------