sge resnet

测试时间

1070 sge_resnet34 416 1 25ms，比原版慢了12ms

1070 sge_resnet34 416 8 90ms，比原版慢了40ms

 

https://github.com/implus/PytorchInsight/blob/bf76edd9f496858686c60fbf6d8a354cc72f913b/classification/models/imagenet/resnet_sge.py

 

import time

import torch.nn as nn
from torch.nn.parameter import Parameter
import torch



__all__ = ['sge_resnet18', 'sge_resnet34', 'sge_resnet50', 'sge_resnet101',
           'sge_resnet152']

class SpatialGroupEnhance(nn.Module):
    def __init__(self, groups = 64):
        super(SpatialGroupEnhance, self).__init__()