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

fix for a bug in distributed training that was introduced lately (#75)

Browse Source
This commit is contained in:
Itai Caspi
2018-11-08 16:52:48 +02:00
committed by GitHub
parent 83e0b09a6a
commit 389c65cbbe
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/graph_managers/graph_manager.py
View File

@@ -195,7 +195,7 @@ class GraphManager(object):
@staticmethod @staticmethod
def create_worker_or_parameters_server(task_parameters: DistributedTaskParameters): def create_worker_or_parameters_server(task_parameters: DistributedTaskParameters):
if task_parameters.framework_type == Frameworks.tensorflow: if task_parameters.framework_type == Frameworks.tensorflow:
GraphManager._create_worker_or_parameters_server_tf(task_parameters) return GraphManager._create_worker_or_parameters_server_tf(task_parameters)
elif task_parameters.framework_type == Frameworks.mxnet: elif task_parameters.framework_type == Frameworks.mxnet:
raise NotImplementedError('Distributed training not implemented for MXNet') raise NotImplementedError('Distributed training not implemented for MXNet')
else: else: