I solved with something more simple:
def squeezeNet_model():
    model_Squeeze = models.squeezenet1_1(pretrained=True)
    num_out = 4
    # change the last conv2d layer
    model_Squeeze.classifier._modules["1"] = nn.Conv2d(512, num_out, kernel_size=(1, 1))


    return model_Squeeze