def forward(self, x):
    features = []
    x1 = self.inc(x)
    features.append(x1)
    x2 = self.down1(x1)
    features.append(x2)
    feats = self.down2(x2)
    features.append(feats)
    feats_down = feats
    for i in range(self.down_num):
        feats_down = nn.MaxPool3d(2)(feats_down)
        features.append(feats_down)
    return (feats, features[::(- 1)])
def forward(self, x):
    features = []
    x1 = self.inc(x)
    features.append(x1)
    x2 = self.down1(x1)
    features.append(x2)
    feats = self.down2(x2)
    features.append(feats)
    feats_down = feats
    for i in range(self.down_num):
        feats_down = nn.MaxPool3d(2)(feats_down)
        features.append(feats_down)
    return (feats, features[::(- 1)])