gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 03:30:19 +01:00
Code Issues Projects Releases Wiki Activity

Changing parameters

Browse Source
This commit is contained in:
Ajay Deshpande
2018-09-14 16:22:13 -07:00
committed by zach dwiel
parent c2991819b4
commit 28926bf2a4
3 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/orchestrators/kubernetes_orchestrator.py
View File

@@ -42,7 +42,7 @@ class Kubernetes(Deploy):
print("Failed to setup redis")
return False
self.deploy_parameters.command += ['-r', self.deploy_parameters.redis_ip, '-p', '{}'.format(self.deploy_parameters.redis_port)]
self.deploy_parameters.command += ['--redis_ip', self.deploy_parameters.redis_ip, '--redis_port', '{}'.format(self.deploy_parameters.redis_port)]
return True

4
rl_coach/orchestrators/test.py
View File

@@ -2,10 +2,10 @@ from rl_coach.orchestrators.kubernetes_orchestrator import KubernetesParameters,
# image = 'gcr.io/constant-cubist-173123/coach:latest'
image = 'ajaysudh/testing:coach'
command = ['python3', 'rl_coach/rollout_worker.py']
command = ['python3', 'rl_coach/rollout_worker.py', '-p', 'CartPole_DQN_distributed']
# command = ['sleep', '10h']
params = KubernetesParameters(image, command, kubeconfig='~/.kube/config', redis_ip='redis-service.ajay.svc', redis_port=6379, num_workers=10)
params = KubernetesParameters(image, command, kubeconfig='~/.kube/config', redis_ip='redis-service.ajay.svc', redis_port=6379, num_workers=1)
# params = KubernetesParameters(image, command, kubeconfig='~/.kube/config')
obj = Kubernetes(params)

12
rl_coach/rollout_worker.py
View File

@@ -20,7 +20,7 @@ def rollout_worker(graph_manager, checkpoint_dir):
task_parameters = TaskParameters()
task_parameters.__dict__['checkpoint_restore_dir'] = checkpoint_dir
graph_manager.create_graph(task_parameters)
graph_manager.phase = RunPhase.TRAIN
graph_manager.act(EnvironmentEpisodes(num_steps=10))
graph_manager.phase = RunPhase.UNDEFINED
@@ -36,11 +36,19 @@ def main():
help='(string) Path to a folder containing a checkpoint to restore the model from.',
type=str,
default='/checkpoint')
parser.add_argument('-r', '--redis_ip',
help="(string) IP or host for the redis server",
default='localhost',
type=str)
parser.add_argument('-rp', '--redis_port',
help="(int) Port of the redis server",
default=6379,
type=int)
args = parser.parse_args()
graph_manager = short_dynamic_import(expand_preset(args.preset), ignore_module_case=True)
graph_manager.agent_parameters.memory.redis_ip = args.redis_ip
graph_manager.agent_params.memory.redis_ip = args.redis_ip
graph_manager.agent_params.memory.redis_port = args.redis_port
rollout_worker(