error while creating model

ooooverflow / BiSeNet

BiSeNet based on pytorch

394 stars 77 forks source link

error while creating model #18

Open sinAshish opened 5 years ago

sinAshish commented 5 years ago

I am trying to run this code, but I get the following error:

 cx = torch.cat((cx1, cx2), dim=1)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 128 and 64 in dimension 2 at /pytorch/aten/src/THC/generic/THCTensorMath.cu:71

Any idea, why is it so?