Fix Task type in automation examples

This commit is contained in:
allegroai 2020-07-04 22:59:24 +03:00
parent 2f5b519cd8
commit 6221c6c096
3 changed files with 3 additions and 3 deletions

View File

@ -3,7 +3,7 @@ from random import sample
from trains import Task
# Connecting TRAINS
task = Task.init(project_name='examples', task_name='Random Hyper-Parameter Search Example')
task = Task.init(project_name='examples', task_name='Random Hyper-Parameter Search Example', task_type=Task.TaskTypes.optimizer)
# Create a hyper-parameter dictionary for the task
params = dict()

View File

@ -2,7 +2,7 @@ from trains import Task
from time import sleep
# Initialize the Task Pipe's first Task used to start the Task Pipe
task = Task.init('examples', 'Simple Controller Task')
task = Task.init('examples', 'Simple Controller Task', task_type=Task.TaskTypes.controller)
# Create a hyper-parameter dictionary for the task
param = dict()

View File

@ -14,7 +14,7 @@ from trains import Task
os.environ["TRAINS_DOCKER_IMAGE"] = "nvidia/cuda --network host"
# initialize TRAINS
task = Task.init(project_name="examples", task_name="Remote Jupyter NoteBook")
task = Task.init(project_name="examples", task_name="Remote Jupyter NoteBook", task_type=Task.TaskTypes.service)
# get rid of all the runtime TRAINS
preserve = (