gitextract_tcky92iz/ ├── .gitignore ├── .travis.yml ├── README.md ├── __init__.py ├── data/ │ ├── __init__.py │ ├── data_generators/ │ │ ├── __init__.py │ │ ├── cifar_generator.py │ │ ├── generator_utils.py │ │ └── mnist_generator.py │ ├── dataset_maps.py │ ├── image_reader.py │ ├── imagenet_augs.py │ └── registry.py ├── hparams/ │ ├── __init__.py │ ├── basic.py │ ├── defaults.py │ ├── lenet.py │ ├── registry.py │ ├── resnet.py │ ├── user.py │ ├── utils.py │ └── vgg.py ├── models/ │ ├── __init__.py │ ├── basic/ │ │ ├── __init__.py │ │ └── basic.py │ ├── lenet/ │ │ ├── __init__.py │ │ └── lenet.py │ ├── registry.py │ ├── resnet/ │ │ ├── __init__.py │ │ └── resnet.py │ ├── utils/ │ │ ├── __init__.py │ │ ├── activations.py │ │ ├── dropouts.py │ │ ├── initializations.py │ │ ├── model_utils.py │ │ └── optimizers.py │ └── vgg/ │ ├── __init__.py │ └── vgg.py ├── requirements.txt ├── scripts/ │ ├── __init__.py │ └── prune/ │ ├── README.md │ ├── __init__.py │ ├── eval.py │ └── prune.py ├── train.py └── training/ ├── __init__.py ├── envs.py ├── flags.py ├── lr_schemes.py └── tpu.py