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

Support DummyModel configuration (Slack https://clearml.slack.com/archives/CTK20V944/p1621469235085400)

Browse Source
This commit is contained in:
allegroai 2021-05-31 15:43:02 +03:00
parent 5e6b4ebe5d
commit 206d86ec5b
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml/model.py
View File

@ -1104,9 +1104,11 @@ class OutputModel(BaseModel):
# noinspection PyProtectedMember
if not self._task._get_model_config_text():
# noinspection PyProtectedMember
task._set_model_config(config_text=model.model_design)
task._set_model_config(
config_text=model.model_design if hasattr(model, "model_design") else model.design.get("design", "")
)
if not self._task.get_labels_enumeration():
task.set_model_label_enumeration(model.data.labels)
task.set_model_label_enumeration(model.data.labels if hasattr(model, "data") else model.labels)
if self._base_model:
self._base_model.update_for_task(