On line 116 of selfsupervised.py, the validation dataset is used for training... is that intentional or is that a mistake?