Files
masks_augments_compare-21/models/transformer_model.py