flyte
flyte copied to clipboard
[Core feature] [plugin] Type - Model analysis using WeightWatcher
Motivation: Why do you think this is important?
For trained models, using WeightWatcher to analyze a deep learning model can help provide default analysis for a model
https://github.com/CalculatedContent/WeightWatcher
Goal: What should the final outcome look like, ideally?
Note: this is a proposal only. May not be the ideal implementation
@task(enable_flytedecks=True)
def train(...) -> Annotated[nn.Module, WeightWatcherRenderer]:
...
Describe alternatives you've considered
self publishing in custom deck.
Propose: Link/Inline OR Additional context
No response
Are you sure this issue hasn't been raised already?
- [X] Yes
Have you read the Code of Conduct?
- [X] Yes