ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-04-30 11:04:23 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add manual model tracking example

Browse Source
This commit is contained in:
allegroai 2020-06-17 01:01:31 +03:00
parent 9ee596f48b
commit a51a1be320
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
examples/reporting/model_config.py
View File

@ -2,7 +2,7 @@
# #
import os import os
from trains import Task from trains import Task, OutputModel
task = Task.init(project_name='examples', task_name='Model configuration example') task = Task.init(project_name='examples', task_name='Model configuration example')
@ -31,6 +31,9 @@ model_config_dict['value'] *= model_config_dict['new value']
labels = {'background': 0, 'cat': 1, 'dog': 2} labels = {'background': 0, 'cat': 1, 'dog': 2}
task.connect_label_enumeration(labels) task.connect_label_enumeration(labels)
# Manually log a local model file
OutputModel().update_weights('my_best_model.bin')
# storing a model: Any saved model (keras / pytorch / tensorflow / etc.) # storing a model: Any saved model (keras / pytorch / tensorflow / etc.)
# will have the task network configuration and label enumeration # will have the task network configuration and label enumeration
print('Any model stored from this point onwards, will contain both model_config and label_enumeration') print('Any model stored from this point onwards, will contain both model_config and label_enumeration')