sum(axis=1, keepdims=True) model.set_state_dict(state_dict) def se_resnext50_32x4d(pretrained=False, num_classes=1000, input_dim=3): model = SENet(SEResNeXtBottleneck, [3, 4, 6, 3], input_dim=input_dim, groups=32, reduction=16, dropout_p=None, in_channels=64, input_3x3=False,...