Cannot batch tensors with different shapes in component 0. First element had shape [224,224,3] and element 2 had shape [2843,2622,3]

Question

I would like to modify this code of SIMCLR with my own dataset containing a directory "train" with two sub-directory:"dog" and "cat",