gcn problem

[foxet]

---

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)