ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 21:33:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add Task.models property

Browse Source
This commit is contained in:
allegroai 2020-03-23 23:25:55 +02:00
parent 0211d233d4
commit 766c8ab24f
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
trains/task.py
View File

@ -458,6 +458,16 @@ class Task(_Task):
artifacts_pairs += list(self._artifacts_manager.registered_artifacts.items()) artifacts_pairs += list(self._artifacts_manager.registered_artifacts.items())
return ReadOnlyDict(artifacts_pairs) return ReadOnlyDict(artifacts_pairs)
@property
def models(self):
# type: () -> Dict[str, List[Model]]
"""
Read-only dictionary of the Task's loaded/stored models
:return: dictionary of models loaded/stored {'input': list(Model), 'output': list(Model)}
"""
return self.get_models()
@classmethod @classmethod
def clone(cls, source_task=None, name=None, comment=None, parent=None, project=None): def clone(cls, source_task=None, name=None, comment=None, parent=None, project=None):
""" """