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

subclass PredictUnit in AutoUnit #545

Closed
wants to merge 1 commit into from

Conversation

JKSenthil
Copy link
Contributor

Summary:

Context

Currently if users want to train, eval, and predict all in one script, they must transfer their modules from AutoUnit to AutoPredictUnit, which is unintuitive and can be error prone

This diff

We subclass in PredictUnit in AutoUnit, which allows users to use the existing predict_step, or override it, when calling predict

Differential Revision: D49441091

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D49441091

JKSenthil added a commit to JKSenthil/tnt that referenced this pull request Sep 20, 2023
Summary:

# Context
Currently if users want to train, eval, and predict all in one script, they must transfer their modules from AutoUnit to AutoPredictUnit, which is unintuitive and can be error prone

# This diff
We subclass in PredictUnit in AutoUnit, which allows users to use the existing predict_step, or override it, when calling `predict`

Differential Revision: D49441091
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D49441091

JKSenthil added a commit to JKSenthil/tnt that referenced this pull request Sep 20, 2023
Summary:

# Context
Currently if users want to train, eval, and predict all in one script, they must transfer their modules from AutoUnit to AutoPredictUnit, which is unintuitive and can be error prone

# This diff
We subclass in PredictUnit in AutoUnit, which allows users to use the existing predict_step, or override it, when calling `predict`

Differential Revision: D49441091
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D49441091

@codecov
Copy link

codecov bot commented Sep 20, 2023

Codecov Report

Merging #545 (f9b963b) into master (ec7636b) will increase coverage by 0.01%.
The diff coverage is 91.66%.

@@            Coverage Diff             @@
##           master     #545      +/-   ##
==========================================
+ Coverage   86.84%   86.86%   +0.01%     
==========================================
  Files          62       62              
  Lines        3931     3943      +12     
==========================================
+ Hits         3414     3425      +11     
- Misses        517      518       +1     
Files Changed Coverage Δ
torchtnt/framework/auto_unit.py 83.72% <91.66%> (+0.38%) ⬆️

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

Summary:

# Context
Currently if users want to train, eval, and predict all in one script, they must transfer their modules from AutoUnit to AutoPredictUnit, which is unintuitive and can be error prone

# This diff
We subclass in PredictUnit in AutoUnit, which allows users to use the existing predict_step, or override it, when calling `predict`

Differential Revision: D49441091
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D49441091

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

Successfully merging this pull request may close these issues.

2 participants