the error occurs while training

[antianxiasimida]

RuntimeError: Given groups=1, weight of size [3, 16, 3, 3], expected input[1, 32, 64, 64] to have 16 channels, but got 32 channels instead

[Fodark]

In the run_training file, in the DATA_PARAMS, you should change IMG_SIZE from 64 to 128, otherwise the generator won't compute the right number of convolutions