references/classification/utils.py
references/segmentation/utils.py

references/classification/train.py
references/segmentation/train.py

references/classification/train_quantization.py
references/segmentation/train.py