Skip to content

Enable resuming with weights.pth saved by visual prompting#2538

Merged
sungchul1 merged 9 commits intoopen-edge-platform:developfrom
sungchul1:vpm-enable-resume
Oct 12, 2023
Merged

Enable resuming with weights.pth saved by visual prompting#2538
sungchul1 merged 9 commits intoopen-edge-platform:developfrom
sungchul1:vpm-enable-resume

Conversation

@sungchul1
Copy link
Copy Markdown
Contributor

@sungchul1 sungchul1 commented Oct 10, 2023

Summary

#2274 (comment)
For this, it was needed to save metas for pytorch lightning in weights.pth.
It was implemented by loading best checkpoint saved in Trainer.checkpoint_callback and saving metas in weights.pth.

How to test

$ tox -vv -e unittest-all-py310-pt1 -- tests/unit/algorithms/visual_prompting/adapters/pytorch_lightning/config/test_visual_prompting_config.py
=============================================================== 7 passed, 1 warning in 0.72s ===============================================================
Name                                                                                                                Stmts   Miss  Cover   Missing
-------------------------------------------------------------------------------------------------------------------------------------------------
src/otx/algorithms/visual_prompting/adapters/pytorch_lightning/config/visual_prompting_config.py                       43      7    84%   56, 102-103, 106, 111-113


$ tox -vv -e unittest-all-py310-pt1 --  tests/unit/algorithms/visual_prompting/tasks/test_inference.py
============================================================= 31 passed, 17 warnings in 24.33s =============================================================
Name                                                                                                                Stmts   Miss  Cover   Missing
-------------------------------------------------------------------------------------------------------------------------------------------------
src/otx/algorithms/visual_prompting/tasks/inference.py                                                                213     28    87%   96, 115-123, 178, 218-219, 224, 337, 347, 352, 387, 434, 438-447, 454-455, 459-460


$ tox -vv -e unittest-all-py310-pt1 --  tests/unit/algorithms/visual_prompting/tasks/test_train.py
============================================================== 1 passed, 10 warnings in 2.22s ==============================================================
Name                                                                                                                Stmts   Miss  Cover   Missing
-------------------------------------------------------------------------------------------------------------------------------------------------
src/otx/algorithms/visual_prompting/tasks/train.py                                                                     41      4    90%   88-89, 96-97


$ tox -vv -e tests-all-py310-pt1 -- tests/integration/cli/visual_prompting/test_visual_prompting.py
=================================================== 8 passed, 2 skipped, 2 warnings in 108.48s (0:01:48) ===================================================

$ tox -vv -e tests-all-py310-pt1 -- tests/e2e/cli/visual_prompting/test_visual_prompting.py
======================================================== 10 passed, 2 warnings in 274.07s (0:04:34) ========================================================

Checklist

  • I have added unit tests to cover my changes.​
  • I have added integration tests to cover my changes.​
  • I have added e2e tests for validation.
  • I have added the description of my changes into CHANGELOG in my target branch (e.g., CHANGELOG in develop).​
  • I have updated the documentation in my target branch accordingly (e.g., documentation in develop).
  • I have linked related issues.

License

  • I submit my code changes under the same Apache License that covers the project.
    Feel free to contact the maintainers if that's a concern.
  • I have updated the license header for each file (see an example below).
# Copyright (C) 2023 Intel Corporation
# SPDX-License-Identifier: Apache-2.0

@sungchul1 sungchul1 added this to the 1.6.0 milestone Oct 10, 2023
@github-actions github-actions bot added ALGO Any changes in OTX Algo Tasks implementation TEST Any changes in tests labels Oct 10, 2023
@sungchul1 sungchul1 marked this pull request as ready for review October 11, 2023 02:19
@sungchul1 sungchul1 requested a review from a team as a code owner October 11, 2023 02:19
@sungchul1 sungchul1 changed the title Enable resuming with weights.pth saved by vpm Enable resuming with weights.pth saved by vpm Oct 11, 2023
@sungchul1 sungchul1 enabled auto-merge (squash) October 11, 2023 02:27
@sungchul1 sungchul1 changed the title Enable resuming with weights.pth saved by vpm Enable resuming with weights.pth saved by visual prompting Oct 11, 2023
Copy link
Copy Markdown
Contributor

@eunwoosh eunwoosh left a comment

Choose a reason for hiding this comment

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

Thanks for your work! I left some questions. Please take a look.

jaegukhyun
jaegukhyun previously approved these changes Oct 12, 2023
@sungchul1 sungchul1 merged commit 26b1a88 into open-edge-platform:develop Oct 12, 2023
@sungchul1 sungchul1 deleted the vpm-enable-resume branch October 12, 2023 05:06
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ALGO Any changes in OTX Algo Tasks implementation TEST Any changes in tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants