Skip to content

fennecinspace/classification-model-search

Folders and files

NameName
Last commit message
Last commit date

Latest commit

5cff54c · Jun 21, 2024

History

1 Commit
Jun 21, 2024
Jun 21, 2024
Jun 21, 2024
Jun 21, 2024
Jun 21, 2024

Repository files navigation

Pytorch Classification Model Search Tool

Do you have a classification dataset and want to know what is the best model to train on it ? You can use this tool to have search for the best classification performance on a specific dataset.

We use a Fire Classification dataset as a use case, Model search is done from this list :

  • VGG16
  • VGG19
  • MobileNet V1
  • MobileNet V2
  • MobileNet V3
  • DenseNet169
  • ResNet50
  • ResNet18
  • Inception V3
  • EfficientNetV2
  • Vit
  • SqueezNet

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published