UNet input size dimension does not match error

[sshuair]

code

import torch
from torchsat.models.segmentation import unet34

model = unet34(2)
inputs = torch.randn((1,3,800,800))
outputs = model(inputs)

error message:

---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-7-62f9f3e6e8e0> in <module>
      1 model = unet34(2)
      2 inputs = torch.randn((1,3,800,800))
----> 3 outputs = model(inputs)

/usr/local/lib/python3.6/site-packages/torch/nn/modules/module.py in __call__(self, *input, **kwargs)
    491             result = self._slow_forward(*input, **kwargs)
    492         else:
--> 493             result = self.forward(*input, **kwargs)
    494         for hook in self._forward_hooks.values():
    495             hook_result = hook(self, input, result)

/usr/local/lib/python3.6/site-packages/torchsat-0.0.1-py3.6.egg/torchsat/models/segmentation/unet.py in forward(self, x)
    172         center = self.center(pool)
    173 
--> 174         dec5 = self.dec5(torch.cat([center, conv5], 1))
    175 
    176         dec4 = self.dec4(torch.cat([dec5, conv4], 1))

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 24 and 25 in dimension 2 at ../aten/src/TH/generic/THTensor.cpp:711
``

[sshuair]

current worked input image size:

63,64,65,66,
127,128,129,130,191,192,193,194,
255,256,257,258,319,320,321,322,383,384,385,386,447,448,449,450,
511,512,513,514,575,576,577,578,639,640,641,642,
703,704,705,706,767,768,769,770,
831,832,833,834,895,896,897,898,
959,960,961,962,
1023,1024,1025,1026,1087,1088,1089,1090,
1151,1152,1153,1154,1215,1216,1217,1218,1279,1280,1281,1282,
1343,1344,1345,1346,1407,1408,1409,1410,1471

[ucalyptus]

@sshuair, did not get you. Is it a Pytorch bug or a repo bug?