search

jcy132 / PatchGCL

Official Pytorch implementation of "Patch-wise Graph Contrastive Learning for Image Translation" (AAAI 2024)
MIT License
6 stars 1 forks source link

RuntimeError: The expanded size of the tensor (2048) must match the existing size (256) at non-singleton dimension 2. Target sizes: [1, 2048, 2048]. Tensor sizes: [1, 256, 256] #3

Open arminbiglari opened 2 months ago

arminbiglari commented 2 months ago

dear author, i am trying to train the net on my own dataset but i face this error. input size 256 * 256 , input_nc = 1

arminbiglari commented 1 month ago

solved: batch size should be 1.