def _forward_func(self, model, **x):
    backbone_feats = model(mode='extract', **x)
    pooling_feats = self.multi_pooling(backbone_feats)
    flat_feats = [xx.view(xx.size(0), (- 1)) for xx in pooling_feats]
    feat_dict = {'feat{}'.format((i + 1)): feat.cpu() for (i, feat) in enumerate(flat_feats)}
    return feat_dict
def _forward_func(self, model, **x):
    backbone_feats = model(mode='extract', **x)
    pooling_feats = self.multi_pooling(backbone_feats)
    flat_feats = [xx.view(xx.size(0), (- 1)) for xx in pooling_feats]
    feat_dict = {'feat{}'.format((i + 1)): feat.cpu() for (i, feat) in enumerate(flat_feats)}
    return feat_dict