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

Fix input model error

Browse Source
This commit is contained in:
allegroai 2022-07-28 18:48:15 +03:00
parent 08af7c664d
commit c58e2551c8
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
clearml/model.py
View File

@ -1069,11 +1069,15 @@ class InputModel(Model):
# noinspection PyProtectedMember
if running_remotely() and (task.is_main_task() or task._is_remote_main_task()):
input_models = task.input_models_id
# noinspection PyBroadException
try:
model_id = next(m_id for m_name, m_id in input_models if m_name == (name or 'Input Model'))
# TODO: (temp fix) At the moment, the UI changes the key of the model hparam
# when modifying its value... There is no way to tell which model was changed
# so just take the first one in case `name` is not in `input_models`
model_id = input_models.get(name, next(iter(input_models.values())))
self._base_model_id = model_id
self._base_model = InputModel(model_id=model_id)._get_base_model()
except StopIteration:
except Exception:
model_id = None
if not model_id: