Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Pass Trainer Arguments into Model.predict #145

Closed
aribornstein opened this issue Feb 24, 2021 · 2 comments
Closed

Pass Trainer Arguments into Model.predict #145

aribornstein opened this issue Feb 24, 2021 · 2 comments
Assignees
@tchaton
Labels
enhancement New feature or request help wanted Extra attention is needed

Comments

@aribornstein
Copy link
Contributor

🚀 Feature

I should have a way to override the predict trainer in model.predict

Motivation

Flashes model.predict doesn't support features like distributed inference this causes pretty significant bottlenecks when evaluating large models we need a way of streamlining this so evaluation will be faster.
@aribornstein aribornstein added enhancement New feature or request help wanted Extra attention is needed labels Feb 24, 2021
@edenlightning edenlightning added this to the 0.2 milestone Mar 22, 2021
@edenlightning edenlightning modified the milestones: 0.2, 0.3 Apr 19, 2021
@edgarriba
Copy link
Contributor

@aribornstein can you describe a bit more the api that would you like here ?

@edenlightning edenlightning removed this from the 0.3 milestone May 13, 2021
@edenlightning
Copy link
Contributor

@aribornstein closing this, reopen with more context if needed

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees

@tchaton tchaton

Labels
enhancement New feature or request help wanted Extra attention is needed
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
4 participants
@aribornstein @edgarriba @tchaton @edenlightning