def forward(self, x):
    low_level_feat = self.low_level_features(x)
    x = self.high_level_features(low_level_feat)
    return (x, low_level_feat)
def forward(self, x):
    low_level_feat = self.low_level_features(x)
    x = self.high_level_features(low_level_feat)
    return (x, low_level_feat)