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.

Sign up for GitHub

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

Jump to bottom

[SPIKE] Explore alternative training frameworks #72

Open
feydan opened this issue Sep 15, 2022 · 0 comments
Open

[SPIKE] Explore alternative training frameworks #72

feydan opened this issue Sep 15, 2022 · 0 comments
Labels
documentation Improvements or additions to documentation duplicate This issue or pull request already exists enhancement New feature or request help wanted Extra attention is needed role: data science / ML engineers
Milestone
2.0 TrashAI Model...

Comments

@feydan
Copy link
Member

feydan commented Sep 15, 2022 •
edited
Loading

Currently we use YoloV5.

Can we try training the model on a different framework?

How does it impact accuracy?

Are there other ways to improve accuracy of the model (i.e. more epochs)?

Might it be useful to have a model selector when using TrashAI?

Acceptance Criteria:

  • After exploring these models, document the findings in a readme file inside of the repo in the model folder. It would be great to see which models were tested, what the findings were, and ideally a way to reproduce the findings.
@epierotti3 epierotti3 moved this to Needs Review in TrashAI Feb 14, 2023
@epierotti3 epierotti3 moved this from Needs Review to In Progress in TrashAI Apr 5, 2023
@epierotti3 epierotti3 added documentation Improvements or additions to documentation enhancement New feature or request help wanted Extra attention is needed role: data science / ML engineers duplicate This issue or pull request already exists labels Apr 5, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
documentation Improvements or additions to documentation duplicate This issue or pull request already exists enhancement New feature or request help wanted Extra attention is needed role: data science / ML engineers
Projects
TrashAI
Status: In Progress
Milestone
2.0 TrashAI Model Improvement
Development

No branches or pull requests
2 participants
@feydan @epierotti3