├── README.md ├── dataset ├── __init__.py ├── imagenet_data.py ├── img2lmdb.py ├── lmdb_dataset.py ├── mk_img_list.py ├── prefetch_data.py └── torchvision_extension.py ├── imagenet_train_cfg.py ├── knowledge_distiller.py ├── models ├── MobileNet.py ├── ResNet.py └── __init__.py ├── tools ├── __init__.py ├── collections.py └── utils.py └── train_with_distillation.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/README.md -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/imagenet_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/dataset/imagenet_data.py -------------------------------------------------------------------------------- /dataset/img2lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/dataset/img2lmdb.py -------------------------------------------------------------------------------- /dataset/lmdb_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/dataset/lmdb_dataset.py -------------------------------------------------------------------------------- /dataset/mk_img_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/dataset/mk_img_list.py -------------------------------------------------------------------------------- /dataset/prefetch_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/dataset/prefetch_data.py -------------------------------------------------------------------------------- /dataset/torchvision_extension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/dataset/torchvision_extension.py -------------------------------------------------------------------------------- /imagenet_train_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/imagenet_train_cfg.py -------------------------------------------------------------------------------- /knowledge_distiller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/knowledge_distiller.py -------------------------------------------------------------------------------- /models/MobileNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/models/MobileNet.py -------------------------------------------------------------------------------- /models/ResNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/models/ResNet.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/collections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/tools/collections.py -------------------------------------------------------------------------------- /tools/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/tools/utils.py -------------------------------------------------------------------------------- /train_with_distillation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bellymonster/Weighted-Soft-Label-Distillation/HEAD/train_with_distillation.py --------------------------------------------------------------------------------