Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Fine Tune Training #1376

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
Merged
BryonLewis merged 17 commits into from
Aug 18, 2025
Merged

Fine Tune Training #1376

BryonLewis merged 17 commits into from
Aug 18, 2025

Conversation

@BryonLewis
Copy link
Collaborator

@BryonLewis BryonLewis commented Sep 13, 2023 •
edited
Loading

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

BryonLewis commented Sep 14, 2023

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

BryonLewis commented Sep 24, 2023

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
@BryonLewis BryonLewis requested a review from naglepuff August 12, 2025 19:02
naglepuff
naglepuff reviewed Aug 13, 2025
View reviewed changes
Copy link
Collaborator

@naglepuff naglepuff left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Full disclosure - I did not get a chance to do any functional testing of this branch, but I did give the code a look, and found some suggestions.

@BryonLewis BryonLewis requested a review from naglepuff August 18, 2025 17:10
@BryonLewis BryonLewis merged commit 40e3968 into main Aug 18, 2025
4 checks passed
@BryonLewis BryonLewis deleted the fine-tune-training branch August 18, 2025 19:25
@BryonLewis BryonLewis mentioned this pull request Sep 10, 2025
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@naglepuff naglepuff naglepuff approved these changes

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

Fine-Tune Training

2 participants

@BryonLewis @naglepuff