Skip to content

Add convenience function to generate inference model from training checkpoint file#285

Merged
wiederm merged 4 commits intomainfrom
dev-load-model-from-chkpt
Oct 19, 2024
Merged

Add convenience function to generate inference model from training checkpoint file#285
wiederm merged 4 commits intomainfrom
dev-load-model-from-chkpt

Conversation

@wiederm
Copy link
Copy Markdown
Member

@wiederm wiederm commented Oct 16, 2024

Pull Request Summary

This PR adds a function that creates a torchscript model from a checkpoint file and sets the right hyperparameters.

Associated Issue(s)

Pull Request Checklist

  • Issue(s) raised/addressed and linked
  • Includes appropriate unit test(s)
  • Appropriate docstring(s) added/updated
  • Appropriate .rst doc file(s) added/updated
  • PR is ready for review

@wiederm wiederm self-assigned this Oct 16, 2024
@wiederm wiederm added the enhancement New feature or request label Oct 16, 2024
@codecov-commenter
Copy link
Copy Markdown

codecov-commenter commented Oct 16, 2024

Codecov Report

❌ Patch coverage is 92.59259% with 2 lines in your changes missing coverage. Please review.
✅ Project coverage is 85.33%. Comparing base (9b3ed45) to head (767d7c5).
⚠️ Report is 809 commits behind head on main.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@wiederm wiederm merged commit e522959 into main Oct 19, 2024
@wiederm wiederm deleted the dev-load-model-from-chkpt branch October 19, 2024 10:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Make it easier to construct inference model from training checkpoint file

2 participants