FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

Fine Tune Training by BryonLewis · Pull Request #1376 · Kitware/dive · GitHub

/ dive Public

Fine Tune Training - #1376

Merged
BryonLewis merged 17 commits into
mainfrom
fine-tune-training
Aug 18, 2025
ConversationCommits (17)ChecksFiles changed
Merged

Fine Tune Training#1376
BryonLewis merged 17 commits into
mainfrom
fine-tune-training

Conversation

BryonLewis commented Sep 13, 2023 •
edited
Loading

Copy link
Copy Markdown
Collaborator

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 commented Sep 14, 2023

Copy link
Copy Markdown
Collaborator Author

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

BryonLewis commented Sep 24, 2023

Copy link
Copy Markdown
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 marked this pull request as ready for review September 24, 2023 18:09
BryonLewis requested a review from naglepuff August 12, 2025 19:02
naglepuff reviewed Aug 13, 2025
View reviewed changes

naglepuff left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

Choose a reason Spam Abuse Off Topic Outdated Duplicate Resolved Low Quality

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.

Comment thread client/platform/desktop/frontend/components/MultiTrainingMenu.vue
Comment thread client/platform/web-girder/views/RunTrainingMenu.vue Outdated
Comment thread client/platform/web-girder/views/RunTrainingMenu.vue Outdated
Comment thread server/dive_server/views_rpc.py Outdated
Comment thread server/dive_server/crud_rpc.py Outdated
Comment thread server/dive_server/crud_rpc.py Outdated
Comment thread server/dive_server/crud_rpc.py Outdated
BryonLewis and others added 3 commits August 14, 2025 14:06
Update client/platform/web-girder/views/RunTrainingMenu.vue
a827543 
Co-authored-by: Michael Nagler <mike.nagler@kitware.com>
Update server/dive_server/crud_rpc.py
6578ba0 
Co-authored-by: Michael Nagler <mike.nagler@kitware.com>
adressing comments
23bd0d2
BryonLewis requested a review from naglepuff August 18, 2025 17:10
BryonLewis merged commit 40e3968 into main Aug 18, 2025
4 checks passed
BryonLewis deleted the fine-tune-training branch August 18, 2025 19:25
BryonLewis mentioned this pull request Sep 10, 2025
Merged
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters
Show hidden characters
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

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

Back | FazBrowse Home | New Git URL