Unanswered
I Have A Training Task That Auto-Magically Saves A Model For Me To Gcs
Hi PanickyMoth78
` torch.save(net.state_dict(), PATH) # auto-uploads to GCS
get all the models from the Task
output_models = Task.current_task().models["output"]
get the last one
last_model = output_models[-1]
set meta-data
last_model.set_metadata(key="my key", value="my value", type="str") `
176 Views
0
Answers
2 years ago
one year ago