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

Predict on fused model #998

Merged
merged 12 commits into from
May 16, 2023
Merged

Predict on fused model #998

merged 12 commits into from
May 16, 2023

Conversation

Louis-Dupont
Copy link
Contributor

@Louis-Dupont Louis-Dupont commented May 14, 2023

Eventually, I took the fused model out of the SgModule to move it to the pipeline, which makes more sense and simplifies the code.
I also added cache with lru_cache to solve the issue of knowing when the pipeline needs to be instantiated or not.

@dagshub
Copy link

dagshub bot commented May 14, 2023

@Louis-Dupont Louis-Dupont marked this pull request as draft May 14, 2023 08:52
@Louis-Dupont Louis-Dupont marked this pull request as ready for review May 14, 2023 10:39
Copy link
Collaborator

@BloodAxe BloodAxe left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
Collaborator

@shaydeci shaydeci left a comment

Choose a reason for hiding this comment

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

LGTM

@Louis-Dupont Louis-Dupont merged commit 363c34a into master May 16, 2023
1 check passed
@Louis-Dupont Louis-Dupont deleted the feature/SG-866-fold_model branch May 16, 2023 14:46
avideci pushed a commit that referenced this pull request May 23, 2023
* predict on fused model

* working version

* fix

* update

* update

* add benchmarl

* add reset

* torch.from_numpy

* fix
avideci pushed a commit that referenced this pull request May 23, 2023
* predict on fused model

* working version

* fix

* update

* update

* add benchmarl

* add reset

* torch.from_numpy

* fix
geoffrey-g-delhomme pushed a commit to geoffrey-g-delhomme/super-gradients that referenced this pull request May 26, 2023
* predict on fused model

* working version

* fix

* update

* update

* add benchmarl

* add reset

* torch.from_numpy

* fix
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants