ycszen / pytorch-segmentation

Pytorch for Segmentation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

gcn problem

foxet opened this issue · comments

commented

TypeError Traceback (most recent call last)
in ()
1 if torch.cuda.is_available():
2 #model = torch.nn.DataParallel(FCN(22))
----> 3 model=FCN(22)
4 model.cuda()

in init(self, num_classes)
81 self.refine10 = Refine(self.num_classes)
82
---> 83 self.out0 = self._classifier(2048)
84 self.out1 = self._classifier(1024)
85 self.out2 = self._classifier(512)

in _classifier(self, inplanes)
95
96 nn.Conv2d(inplanes, inplanes, 3, padding=1, bias=False),
---> 97 nn.BatchNorm2d(inplanes/2),
98 nn.ReLU(inplace=True),
99 nn.Dropout(.1),

/home/omnisky/anaconda3/lib/python3.6/site-packages/torch/nn/modules/batchnorm.py in init(self, num_features, eps, momentum, affine)
16 self.momentum = momentum
17 if self.affine:
---> 18 self.weight = Parameter(torch.Tensor(num_features))
19 self.bias = Parameter(torch.Tensor(num_features))
20 else:

TypeError: torch.FloatTensor constructor received an invalid combination of arguments - got (float), but expected one of:

  • no arguments
  • (int ...)
    didn't match because some of the arguments have invalid types: (float)
  • (torch.FloatTensor viewed_tensor)
    didn't match because some of the arguments have invalid types: (float)
  • (torch.Size size)
    didn't match because some of the arguments have invalid types: (float)
  • (torch.FloatStorage data)
    didn't match because some of the arguments have invalid types: (float)
  • (Sequence data)
    didn't match because some of the arguments have invalid types: (float)