-
Notifications
You must be signed in to change notification settings - Fork 1
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
Task Label Groups Support #22
Conversation
Test Results 6 files 6 suites 1h 5m 7s ⏱️ Results for commit 5f6f0db. ♻️ This comment has been updated with latest results. |
☂️ Python Coverage
Overall Coverage
New FilesNo new covered files... Modified Files
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Leaving comments for now in case multi-task support is still needed
@@ -425,7 +426,8 @@ def anchors_from_dataset( | |||
|
|||
widths = [] | |||
inputs = None | |||
for inp, labels in loader: | |||
for inp, task_labels in loader: | |||
labels = next(iter(task_labels.values())) # TODO: handle multiple tasks |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
# TODO: handle multiple tasks
What is needed for this TODO?
Co-authored-by: conorsim <[email protected]>
* handling SIGTERM signal * resume argument takes path * basic task group labels support * updated requirements * fixed tests * fixed loader test * Update luxonis_train/models/luxonis_model.py Co-authored-by: conorsim <[email protected]> --------- Co-authored-by: conorsim <[email protected]>
Added support for task label groups as implemented in
luxonis-ml
in 101.