ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Allow registering offline models

Browse Source
This commit is contained in:
Alex Burlacu
2023-03-23 18:32:53 +02:00
parent a5d25b1a88
commit dc5be02328
3 changed files with 77 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
clearml/task.py
View File

@@ -2930,6 +2930,11 @@ class Task(_Task):
StorageManager.upload_file(local_file=local_file.as_posix(), remote_url=artifact.uri)
# noinspection PyProtectedMember
task_holding_reports._edit(execution=current_task.data.execution)
for output_model in export_data.get("offline_output_models", []):
model = OutputModel(task=current_task, **output_model["init"])
if output_model.get("output_uri"):
model.set_upload_destination(output_model.get("output_uri"))
model.update_weights(auto_delete_file=False, **output_model["weights"])
# logs
TaskHandler.report_offline_session(task_holding_reports, session_folder, iteration_offset=iteration_offset)
# metrics
@@ -3833,6 +3838,9 @@ class Task(_Task):
if self._offline_mode and not is_sub_process:
# noinspection PyBroadException
try:
# make sure the state of the offline data is saved
self._edit()
# create zip file
offline_folder = self.get_offline_mode_folder()
zip_file = offline_folder.as_posix() + '.zip'