Requires:
- Hydra (https://github.com/facebookresearch/hydra)
- PyTorch
- Seaborn
Regular training: run python train.py
Student Teacher training: run python main.py
Requires:
Regular training: run python train.py
Student Teacher training: run python main.py