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

Fine Tune Training #1376

Open
wants to merge 11 commits into
base: main
Choose a base branch
from
Open

Fine Tune Training #1376

wants to merge 11 commits into from

Conversation

BryonLewis
Copy link
Collaborator

@BryonLewis BryonLewis commented Sep 13, 2023

Will allow users to choose a model to allow for fine-tune training.
resolves #1375

  • Logs the model files that are available after downloading the add-ons for the web version
  • The endpoint for training configs has been updated to show the models as well as the configs.

TODO:

  • Update front-end to utilize modified Training configs
  • Add optional checkbox on front-end to fine-tune model
  • Update the endpoint to run training so that a model can be passed in as well
  • Add in argument to the training task in celery to utilize new input

Desktop Version

  • Find a way to search and find available models in the desktop version
  • Update the interface to show the models that can be fine tuned
  • Update the call to running training to add in the new optional fine-tune training model.

@BryonLewis
Copy link
Collaborator Author

Everything is templated out for the fine-tune training just need some executables (docker container and VIAME desktop) to test with.

@BryonLewis
Copy link
Collaborator Author

Tested the following:

  • Web version will display the --init-weights command when configured in the UI.
  • Desktop version does the same and displays --init-weights.

Possibly Todo:

  • Better arrangement of values (like dividing into system models and user models)
  • Styling for the training menu so it expands better, possibility having a max height and scrolling.

@BryonLewis BryonLewis marked this pull request as ready for review September 24, 2023 18:09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Fine-Tune Training
1 participant