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

[aniket1493]

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