The library comes in two parts:
- distillation.labeller
- distillation.utils
- distillation.dataloaders
The basic idea is that you have a trained model that can be loaded in PyTorch.
If you do, you call labeller.label_data(data_loader, checkpoint_filename),
and the