Hi, I'm trying to run experiment "train_poisson_gradcomp_img.py"
I'm using two 256x256 B&W images, but when I run the script I keep getting:
Traceback (most recent call last):
File "/content/siren/experiment_scripts/train_poisson_gradcomp_img.py", line 44, in
sidelength=512)
File "/content/siren/dataio.py", line 854, in init
paddedImg[:, 512 - 340:512, :] = self.img2
RuntimeError: The expanded size of the tensor (84) must match the existing size (256) at non-singleton dimension 1. Target sizes: [1, 84, 256]. Tensor sizes: [256, 256]
Hi, I'm trying to run experiment "train_poisson_gradcomp_img.py" I'm using two 256x256 B&W images, but when I run the script I keep getting:
Traceback (most recent call last): File "/content/siren/experiment_scripts/train_poisson_gradcomp_img.py", line 44, in
sidelength=512)
File "/content/siren/dataio.py", line 854, in init
paddedImg[:, 512 - 340:512, :] = self.img2
RuntimeError: The expanded size of the tensor (84) must match the existing size (256) at non-singleton dimension 1. Target sizes: [1, 84, 256]. Tensor sizes: [256, 256]
Any clue on how to fix? Many thanks in advance