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

Fix example Windows support

Browse Source
This commit is contained in:
allegroai 2019-10-06 21:57:50 +03:00
parent 0a5696235b
commit 7677f9dccd
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/manual_model_config.py
View File

@ -1,5 +1,8 @@
# TRAINS - Example of manual model configuration # TRAINS - Example of manual model configuration
# #
import os
from tempfile import gettempdir
import torch import torch
from trains import Task from trains import Task
@ -25,5 +28,6 @@ task.set_model_label_enumeration({'background': 0, 'cat': 1, 'dog': 2})
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')
# storing the model, it will have the task network configuration and label enumeration # storing the model, it will have the task network configuration and label enumeration
torch.save(model, '/tmp/model')
torch.save(model, os.path.join(gettempdir(), "model"))
print('Model saved') print('Model saved')

3
examples/requirements.txt
View File

@ -1,7 +1,8 @@
absl-py>=0.7.1 absl-py>=0.7.1
Keras>=2.2.4 Keras>=2.2.4
joblib>=0.13.2 joblib>=0.13.2
matplotlib>=3.1.1 matplotlib>=3.1.1 ; python_version >= '3.6'
matplotlib == 3.0.3 ; python_version < '3.6'
seaborn>=0.9.0 seaborn>=0.9.0
sklearn>=0.0 sklearn>=0.0
tensorboard>=1.14.0 tensorboard>=1.14.0