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.

Sign up for GitHub

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

Jump to bottom

Option to only return the last hidden layer output from models #342

Open
danieldk opened this issue Oct 3, 2023 · 0 comments
Open

Option to only return the last hidden layer output from models #342

danieldk opened this issue Oct 3, 2023 · 0 comments
Labels
feat/model Feature: models type/feature Type: Feature
Milestone
v2.0.0

Comments

@danieldk
Copy link
Collaborator

danieldk commented Oct 3, 2023

In many applications we only need the last layer and letting go of references to intermediate layers can save some memory during inference.
@danieldk danieldk added type/feature Type: Feature feat/model Feature: models labels Oct 3, 2023
@danieldk danieldk modified the milestones: v1.4.0, v2.0.0 Oct 3, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
feat/model Feature: models type/feature Type: Feature
Projects
None yet
Milestone
v2.0.0
Development

No branches or pull requests
1 participant
@danieldk