├── README.md ├── configs └── default_config.yaml ├── data ├── download_checkpoint_4steps.sh └── download_coco_train2014.sh ├── dataset.py ├── main.py ├── requirements.txt ├── run_inference.ipynb ├── run_training.sh ├── solver.py ├── stuff └── imgs.jpg ├── train.py ├── train_utils.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/README.md -------------------------------------------------------------------------------- /configs/default_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/configs/default_config.yaml -------------------------------------------------------------------------------- /data/download_checkpoint_4steps.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/data/download_checkpoint_4steps.sh -------------------------------------------------------------------------------- /data/download_coco_train2014.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/data/download_coco_train2014.sh -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/dataset.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_inference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/run_inference.ipynb -------------------------------------------------------------------------------- /run_training.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/run_training.sh -------------------------------------------------------------------------------- /solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/solver.py -------------------------------------------------------------------------------- /stuff/imgs.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/stuff/imgs.jpg -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/train.py -------------------------------------------------------------------------------- /train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/train_utils.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/quickjkee/YSDA-Consistency-Distillation/HEAD/utils.py --------------------------------------------------------------------------------