/simclr/
../
main.py
models.py