flyte icon indicating copy to clipboard operation
flyte copied to clipboard

[Core feature] [plugin] Type - Model analysis using WeightWatcher

Open kumare3 opened this issue 1 year ago • 2 comments

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

Onboarding

kumare3 avatar Oct 22 '22 20:10 kumare3