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
Looks to be working 🚀 just need to test one more thing. Thank you CostlyOstrich36
Can you please clarify? What does torch.nn.Module
hold within? Isn't it just the weights configuration?
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