vsitzmann / siren

Official implementation of "Implicit Neural Representations with Periodic Activation Functions"
MIT License
1.72k stars 247 forks source link

Runtime error in experiment poisson_gradcomp #34

Closed smithee77 closed 3 years ago

smithee77 commented 3 years ago

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