ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-01 01:26:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix task.update_output_model() should always upload local models to a remote server

Browse Source
This commit is contained in:
allegroai 2022-11-11 23:46:20 +02:00
parent ebe7aa3fb7
commit 7f6c80f292
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
clearml/backend_interface/task/task.py
View File

@ -842,6 +842,9 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
then ClearML updates the model object associated with the Task an API call. The API call uses with the URI
of the uploaded file, and other values provided by additional arguments.
Notice: A local model file will be uploaded to the task's `output_uri` destination,
If no `output_uri` was specified, the default files-server will be used to store the model file/s.
:param model_path: A local weights file or folder to be uploaded.
If remote URI is provided (e.g. http:// or s3: // etc) then the URI is stored as is, without any upload
:param name: The updated model name.
@ -859,6 +862,7 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
:return: The URI of the uploaded weights file.
Notice: upload is done is a background thread, while the function call returns immediately
"""
output_uri = self.storage_uri or self._get_default_report_storage_uri()
from ...model import OutputModel
output_model = OutputModel(
task=self,
@ -868,7 +872,10 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
)
output_model.connect(task=self, name=name)
url = output_model.update_weights(
weights_filename=model_path, iteration=iteration, auto_delete_file=auto_delete_file
weights_filename=model_path,
upload_uri=output_uri,
iteration=iteration,
auto_delete_file=auto_delete_file
)
return url