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

Allow auto-model logging callbacks to drop logging

Browse Source
This commit is contained in:
allegroai 2020-06-16 22:01:52 +03:00
parent 261d4aa958
commit e03d421b46
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
trains/binding/frameworks/__init__.py
View File

@ -167,6 +167,11 @@ class WeightsFileHandler(object):
except Exception: except Exception:
pass pass
# if callback forced us to leave they return None
if model_info is None:
# callback forced quit
return filepath
if not model_info.local_model_path: if not model_info.local_model_path:
get_logger(TrainsFrameworkAdapter).debug("Could not retrieve model file location, model is not logged") get_logger(TrainsFrameworkAdapter).debug("Could not retrieve model file location, model is not logged")
return filepath return filepath
@ -324,6 +329,13 @@ class WeightsFileHandler(object):
model_info = cb('save', model_info) model_info = cb('save', model_info)
except Exception: except Exception:
pass pass
# if callbacks force us to leave they return None
if model_info is None:
# callback forced quit
return saved_path
# update the trains_out_model after the pre callbacks
trains_out_model = model_info.model trains_out_model = model_info.model
# check if object already has InputModel # check if object already has InputModel