I think this would be right up your alley 🙂
https://clear.ml/docs/latest/docs/references/sdk/model_outputmodel
And you can update weights later using:
https://clear.ml/docs/latest/docs/references/sdk/model_outputmodel#update_weights

Looks to be working 🚀 just need to test one more thing. Thank you CostlyOstrich36

Hey having a few issues with this

I can run clearml.OutputModel(task, framework='pytorch') to get the model from a previous task. but how can I get the pytorch model ( torch.nn.Module ) from the output model object

RobustRat47 , do you mean the weights file?

the Model object

Can you please clarify? What does torch.nn.Module hold within? Isn't it just the weights configuration?