gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-03-02 06:35:47 +01:00
Code Issues Projects Releases Wiki Activity

Adding parameteres, checking transitions before training

Browse Source
This commit is contained in:
Ajay Deshpande
2018-10-08 14:49:46 -07:00
committed by zach dwiel
parent 0f46877d7e
commit b285a02023
5 changed files with 18 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
rl_coach/orchestrators/kubernetes_orchestrator.py
View File

@@ -3,6 +3,7 @@ import uuid
import json
import time
from typing import List
from configparser import ConfigParser, Error
from rl_coach.orchestrators.deploy import Deploy, DeployParameters
from kubernetes import client as k8sclient, config as k8sconfig
from rl_coach.memories.backend.memory import MemoryBackendParameters
@@ -100,8 +101,8 @@ class Kubernetes(Deploy):
if not trainer_params:
return False
trainer_params.command += ['--memory_backend_params', json.dumps(self.params.memory_backend_parameters.__dict__)]
trainer_params.command += ['--data_store_params', json.dumps(self.params.data_store_params.__dict__)]
trainer_params.command += ['--memory-backend-params', json.dumps(self.params.memory_backend_parameters.__dict__)]
trainer_params.command += ['--data-store-params', json.dumps(self.params.data_store_params.__dict__)]
name = "{}-{}".format(trainer_params.run_type, uuid.uuid4())
@@ -176,8 +177,8 @@ class Kubernetes(Deploy):
if not worker_params:
return False
worker_params.command += ['--memory_backend_params', json.dumps(self.params.memory_backend_parameters.__dict__)]
worker_params.command += ['--data_store_params', json.dumps(self.params.data_store_params.__dict__)]
worker_params.command += ['--memory-backend-params', json.dumps(self.params.memory_backend_parameters.__dict__)]
worker_params.command += ['--data-store-params', json.dumps(self.params.data_store_params.__dict__)]
name = "{}-{}".format(worker_params.run_type, uuid.uuid4())

8
rl_coach/orchestrators/start_training.py
View File

@@ -77,19 +77,19 @@ if __name__ == '__main__':
parser.add_argument('-ns', '--nfs-server',
help="(string) Addresss of the nfs server.",
type=str,
required=True)
required=False)
parser.add_argument('-np', '--nfs-path',
help="(string) Exported path for the nfs server.",
type=str,
required=True)
required=False)
parser.add_argument('--s3-end-point',
help="(string) S3 endpoint to use when S3 data store is used.",
type=str,
required=True)
required=False)
parser.add_argument('--s3-bucket-name',
help="(string) S3 bucket name to use when S3 data store is used.",
type=str,
required=True)
required=False)
parser.add_argument('--num-workers',
help="(string) Number of rollout workers",
type=int,