pyTorch image classifier models

model

• ResNet-based

  • ResNet 18
  • ResNet 50
  • ResNet 101

• To Do

Datasets

All data has the same resolution of 32x32

• MNIST
• FMNIST
• CelebA
  • Just use gender information so this is binary classification
• Cifar10