mirror of
https://github.com/clearml/clearml
synced 2025-01-31 17:17:00 +00:00
Fix example Windows support
This commit is contained in:
parent
0a5696235b
commit
7677f9dccd
@ -1,5 +1,8 @@
|
||||
# TRAINS - Example of manual model configuration
|
||||
#
|
||||
import os
|
||||
from tempfile import gettempdir
|
||||
|
||||
import torch
|
||||
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')
|
||||
|
||||
# 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')
|
||||
|
@ -1,7 +1,8 @@
|
||||
absl-py>=0.7.1
|
||||
Keras>=2.2.4
|
||||
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
|
||||
sklearn>=0.0
|
||||
tensorboard>=1.14.0
|
||||
|
Loading…
Reference in New Issue
Block a user