CCNet icon indicating copy to clipboard operation
CCNet copied to clipboard

About ccnet.py. 'return [x, x_dsn]'

Open yearing1017 opened this issue 5 years ago • 1 comments

def forward(self, x, recurrence=1):
        x = self.relu1(self.bn1(self.conv1(x)))
        x = self.relu2(self.bn2(self.conv2(x)))
        x = self.relu3(self.bn3(self.conv3(x)))
        x = self.maxpool(x)
        x = self.layer1(x)
        x = self.layer2(x)
        x = self.layer3(x)
        x_dsn = self.dsn(x)
        x = self.layer4(x)
        x = self.head(x, recurrence)
        return [x, x_dsn]
  • I do not know why return [x, x_dsn]? why not return (torch.cat(x, x_dsn))

yearing1017 avatar Mar 28 '20 02:03 yearing1017

@yearing1017 It's for deep supervision. Please refer to the loss function.

speedinghzl avatar Aug 03 '20 09:08 speedinghzl