jahongir7174 / deeplab-pt Goto Github PK

def fuse(self):
    for n, m in self.named_modules():
        if type(m) is Conv and hasattr(m, 'norm'):
            if isinstance(m.relu, torch.nn.SiLU):
                m.relu = SiLU(inplace=True)
            m.conv = fuse_conv(m.conv, m.norm)
            delattr(m, 'norm')
            m.forward = m.fuse_forward
        elif n == 'fc':
            m[1] = SiLU(inplace=True)
    return self