Skip to content

PyTorch Models

Jump to bottom
Thanos Masouris edited this page Sep 3, 2022 · 1 revision

We opted to quantize PyTorch models, pre-trained on CIFAR-10 for classification. These models were obtained from the chenyaofo/pytorch-cifar-models repository on GitHub. In particular, we used the following models:

  • ResNet20 (resnet20)
  • VGG16 (vgg16_bn)
  • MobileNetV2 (mobilenetv2_x1_4)
  • ShuffleNetV2 (shufflenetv2_x2_0)
  • RepVGG (repvgg_a2)

Table of Contents

Clone this wiki locally