├── README.md ├── dataloader ├── __pycache__ │ ├── autoaugment.cpython-38.pyc │ ├── cutout.cpython-38.pyc │ ├── dataloader.cpython-38.pyc │ ├── dataloaders.cpython-38.pyc │ ├── ops.cpython-38.pyc │ ├── randaug.cpython-38.pyc │ └── random_erase.cpython-38.pyc ├── autoaugment.py ├── cutout.py ├── dataloaders.py ├── randaug.py └── random_erase.py ├── main.py ├── main_mixskd_cifar.py ├── main_mixskd_imagenet.py ├── methods ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── adversarial.cpython-38.pyc │ ├── bake.cpython-38.pyc │ ├── byot.cpython-38.pyc │ ├── cs_kd.cpython-38.pyc │ ├── cutmix.cpython-38.pyc │ ├── ddsgd.cpython-38.pyc │ ├── dks.cpython-38.pyc │ ├── focal_loss.cpython-38.pyc │ ├── frskd.cpython-38.pyc │ ├── label_smooth.cpython-38.pyc │ ├── manifold_mixup.cpython-38.pyc │ ├── mixup.cpython-38.pyc │ ├── ps_kd.cpython-38.pyc │ ├── sad.cpython-38.pyc │ └── tf_kd_reg.cpython-38.pyc ├── adversarial.py ├── bake.py ├── byot.py ├── cs_kd.py ├── cutmix.py ├── ddsgd.py ├── dks.py ├── focal_loss.py ├── frskd.py ├── label_smooth.py ├── manifold_mixup.py ├── mixup.py ├── ps_kd.py ├── sad.py └── tf_kd_reg.py ├── models ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── bifpnc.cpython-38.pyc │ ├── resnet.cpython-38.pyc │ ├── resnet_imagenet.cpython-38.pyc │ ├── resnet_imagenet_mixskd.cpython-38.pyc │ ├── resnet_mixskd.cpython-38.pyc │ └── utils.cpython-38.pyc ├── bifpnc.py ├── resnet.py ├── resnet_imagenet.py ├── resnet_imagenet_mixskd.py ├── resnet_mixskd.py └── utils.py ├── scripts ├── main_cifar.sh ├── main_mixskd_imagenet.sh ├── reference.md └── segmentation │ ├── ade20k.sh │ ├── coco_stuff_164K.sh │ └── pascal_voc.sh └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/README.md -------------------------------------------------------------------------------- /dataloader/__pycache__/autoaugment.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/dataloader/__pycache__/autoaugment.cpython-38.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/cutout.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/dataloader/__pycache__/cutout.cpython-38.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/dataloader.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/dataloader/__pycache__/dataloader.cpython-38.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/dataloaders.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/dataloader/__pycache__/dataloaders.cpython-38.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/ops.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/dataloader/__pycache__/ops.cpython-38.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/randaug.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/dataloader/__pycache__/randaug.cpython-38.pyc -------------------------------------------------------------------------------- /dataloader/__pycache__/random_erase.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/dataloader/__pycache__/random_erase.cpython-38.pyc -------------------------------------------------------------------------------- /dataloader/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/dataloader/autoaugment.py -------------------------------------------------------------------------------- /dataloader/cutout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/dataloader/cutout.py -------------------------------------------------------------------------------- /dataloader/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/dataloader/dataloaders.py -------------------------------------------------------------------------------- /dataloader/randaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/dataloader/randaug.py -------------------------------------------------------------------------------- /dataloader/random_erase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/dataloader/random_erase.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/main.py -------------------------------------------------------------------------------- /main_mixskd_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/main_mixskd_cifar.py -------------------------------------------------------------------------------- /main_mixskd_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/main_mixskd_imagenet.py -------------------------------------------------------------------------------- /methods/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__init__.py -------------------------------------------------------------------------------- /methods/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/adversarial.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/adversarial.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/bake.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/bake.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/byot.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/byot.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/cs_kd.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/cs_kd.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/cutmix.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/cutmix.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/ddsgd.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/ddsgd.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/dks.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/dks.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/focal_loss.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/focal_loss.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/frskd.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/frskd.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/label_smooth.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/label_smooth.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/manifold_mixup.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/manifold_mixup.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/mixup.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/mixup.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/ps_kd.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/ps_kd.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/sad.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/sad.cpython-38.pyc -------------------------------------------------------------------------------- /methods/__pycache__/tf_kd_reg.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/__pycache__/tf_kd_reg.cpython-38.pyc -------------------------------------------------------------------------------- /methods/adversarial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/adversarial.py -------------------------------------------------------------------------------- /methods/bake.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/bake.py -------------------------------------------------------------------------------- /methods/byot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/byot.py -------------------------------------------------------------------------------- /methods/cs_kd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/cs_kd.py -------------------------------------------------------------------------------- /methods/cutmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/cutmix.py -------------------------------------------------------------------------------- /methods/ddsgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/ddsgd.py -------------------------------------------------------------------------------- /methods/dks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/dks.py -------------------------------------------------------------------------------- /methods/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/focal_loss.py -------------------------------------------------------------------------------- /methods/frskd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/frskd.py -------------------------------------------------------------------------------- /methods/label_smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/label_smooth.py -------------------------------------------------------------------------------- /methods/manifold_mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/manifold_mixup.py -------------------------------------------------------------------------------- /methods/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/mixup.py -------------------------------------------------------------------------------- /methods/ps_kd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/ps_kd.py -------------------------------------------------------------------------------- /methods/sad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/sad.py -------------------------------------------------------------------------------- /methods/tf_kd_reg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/methods/tf_kd_reg.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/bifpnc.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/__pycache__/bifpnc.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/__pycache__/resnet.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet_imagenet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/__pycache__/resnet_imagenet.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet_imagenet_mixskd.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/__pycache__/resnet_imagenet_mixskd.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet_mixskd.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/__pycache__/resnet_mixskd.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /models/bifpnc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/bifpnc.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resnet_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/resnet_imagenet.py -------------------------------------------------------------------------------- /models/resnet_imagenet_mixskd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/resnet_imagenet_mixskd.py -------------------------------------------------------------------------------- /models/resnet_mixskd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/resnet_mixskd.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/models/utils.py -------------------------------------------------------------------------------- /scripts/main_cifar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/scripts/main_cifar.sh -------------------------------------------------------------------------------- /scripts/main_mixskd_imagenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/scripts/main_mixskd_imagenet.sh -------------------------------------------------------------------------------- /scripts/reference.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/scripts/reference.md -------------------------------------------------------------------------------- /scripts/segmentation/ade20k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/scripts/segmentation/ade20k.sh -------------------------------------------------------------------------------- /scripts/segmentation/coco_stuff_164K.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/scripts/segmentation/coco_stuff_164K.sh -------------------------------------------------------------------------------- /scripts/segmentation/pascal_voc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/scripts/segmentation/pascal_voc.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/winycg/Self-KD-Lib/HEAD/utils.py --------------------------------------------------------------------------------