Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Pokemon Classifier #2

Closed
AgastyaPatel opened this issue Jan 4, 2024 · 2 comments
Closed

Pokemon Classifier #2

AgastyaPatel opened this issue Jan 4, 2024 · 2 comments
Labels
Model-Training-History This tag contains issues which would contain history of the progress for the model training process

Comments

@AgastyaPatel
Copy link
Owner

AgastyaPatel commented Jan 4, 2024

Pokemon Classifier:
Working on pokemon classifier along with making some comparisons between models, effect of learning rate and epoch.

Model Deployed: Hugging Face Spaces

@AgastyaPatel
Copy link
Owner Author

Commit: eb1a75e

  • Resnet 18 model
  • Resnet 18 with good learning rate
  • Comparing training performance of the models also with having different epochs
    Issue: Model has high accuracy with downloaded dataset and seems to underperform with real world use case

Commit: f008400

  • Added Resnet34, Resnet34 with good learning rate
  • Fixed accuracy of Resnet18 models with data-augmentation

@AgastyaPatel AgastyaPatel added the Model-Training-History This tag contains issues which would contain history of the progress for the model training process label Jan 4, 2024
@AgastyaPatel AgastyaPatel added this to the Training History milestone Jan 4, 2024
AgastyaPatel added a commit that referenced this issue Jan 4, 2024
Also fixed naming in performance graph
@AgastyaPatel
Copy link
Owner Author

Closing the project as trying multiple pretrained models like resnet18, resnet34, resnet50 and convnext_nano have all resulted in similar accuracy.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Model-Training-History This tag contains issues which would contain history of the progress for the model training process
Projects
None yet
Development

No branches or pull requests

1 participant