the error occurs while training

dvl-tum / ciagan

Official PyTorch implementation of CIAGAN

MIT License

76 stars 26 forks source link

the error occurs while training #8

Open antianxiasimida opened 3 years ago

antianxiasimida commented 3 years ago

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 commented 3 years ago

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