code changes required for image size128

[loseway12138]

Hi,
I wanted to run the code for images of size 128 as input instead of 64.
I see that image size 64 is hardcoded in "train.py" while transforming into tensors.

Is there any other place across the code where I need to change the image size (or other parameters dependent on image size) to run it on 128 size input image.

[woozzu]

@loseway12138 You can change the resolution by modifying the data loader and the model definition (in model.py).