Pretrained Resnet used for SED

pretrained_resnet3d.py

class PretrainedResNet(torch.nn.Module):

    """Docstring for PretrainedResNet. """

    def __init__(self, outputdim):
        """TODO: to be defined1.

        :outputdim: TODO

        """
        torch.nn.Module.__init__(self)

        import pretorched  # HUIIIIIII
        self._outputdim = outputdim
        model_name = 'resnet3d50'
        self.net = pretorched.__dict__[model_name](
            num_classes=339, pretrained='moments')
        self.net.last_linear = nn.Linear(2048, outputdim)
        self.net.avgpool = nn.AdaptiveAvgPool3d((None, 1, 1))

    def forward(self, x):
        x = self.net.conv1(x)
        x = self.net.bn1(x)
        x = self.net.relu(x)
        x = self.net.maxpool(x)
        x = self.net.layer1(x)
        x = self.net.layer2(x)
        x = self.net.layer3(x)
        x = self.net.layer4(x)
        x = self.net.avgpool(x)
        x = x.transpose(1, 2).contiguous()
        x = x.view(x.shape[0], x.shape[1], -1)
        return self.net.last_linear(x)