Models and pre-trained weights - PyTorch
pytorch.org › vision › masterObtaining a pre-trained quantized model can be done with a few lines of code: import torchvision.models as models model = models.quantization.mobilenet_v2(pretrained=True, quantize=True) model.eval() # run the model with quantized inputs and weights out = model(torch.rand(1, 3, 224, 224))