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

Fix tracking model bug that pinned n_filters, encoder_dim and embedding_dim to 64 #606

Merged
merged 4 commits into from
Jun 4, 2022

Conversation

vanvalen
Copy link
Contributor

@vanvalen vanvalen commented Jun 4, 2022

What

  • Fixed a bug that required the tracking model to have n_filters, encoder_dim, and embedding_dim be pinned to 64

Why

  • Model optimization is going to require us to change these parameters to improve performance and reduce model size. This pull request makes that substantially easier by fixing this bug.

Copy link
Member

@msschwartz21 msschwartz21 left a comment

Choose a reason for hiding this comment

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

Can you leave a note in this PR thread about what new kwargs would be passed into deepcell.data.tracking.Track?

@msschwartz21 msschwartz21 added the fix Fixes a bug label Jun 4, 2022
@@ -53,7 +53,8 @@
class Track(object): # pylint: disable=useless-object-inheritance

def __init__(self, path=None, tracked_data=None,
appearance_dim=32, distance_threshold=64):
appearance_dim=32, distance_threshold=64,
**kwargs):
Copy link
Member

Choose a reason for hiding this comment

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

Why are kwargs being exposed here? Is this coming from the other PR to expose crop mode and norm?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Yeah I think that's a hold over from the other PR.

@msschwartz21 msschwartz21 changed the title Dvv/fix tracking model bug Fix tracking model bug that pinned n_filters, encoder_dim and embedding_dim to 64 Jun 4, 2022
@msschwartz21 msschwartz21 merged commit a251234 into master Jun 4, 2022
@msschwartz21 msschwartz21 deleted the dvv/fix-tracking-model-bug branch June 4, 2022 02:33
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
fix Fixes a bug
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants