search

knazeri / edge-informed-sisr

Edge-Informed Single Image Super-Resolution, ICCVW 2019 https://arxiv.org/abs/1909.05305
Other
78 stars 20 forks source link

invalid argument 0: Sizes of tensors must match except in dimension 1. Got 256 and 130 in dimension 2 at /pytorch/aten/src/THC/generic/THCTensorMath.cu:71 #15

Open aniket1493 opened 3 years ago

aniket1493 commented 3 years ago

Training epoch: 1 Traceback (most recent call last): File "main.py", line 118, in main() File "main.py", line 48, in main model.train() File "/workspace/aniket/edge-informed-sisr/src/edge_match.py", line 120, in train hr_images_pred, gen_loss, dis_loss, logs = self.sr_model.process(lr_images, hr_images, lr_edges, hr_edges_pred) File "/workspace/aniket/edge-informed-sisr/src/models.py", line 210, in process outputs = self(lr_images, hr_edges) File "/workspace/aniket/edge-informed-sisr/eisr-env/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, **kwargs) File "/workspace/aniket/edge-informed-sisr/src/models.py", line 263, in forward inputs = torch.cat((hr_images, hr_edges), dim=1) RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 256 and 130 in dimension 2 at /pytorch/aten/src/THC/generic/THCTensorMath.cu:71