models/official/projects/simclr/tasks
..
simclr.py