imnet_resnet50_scratch/transforms.py
transforms_v2.py