Fix cluster definition of chief worker

OpenNMT · Nov 16, 2017 · 5a4d715 · 5a4d715
1 parent 4e7d6dd
commit 5a4d715
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bin/main.py b/bin/main.py
@@ -229,7 +229,7 @@ def main():
   if args.chief_host:
     os.environ["TF_CONFIG"] = json.dumps({
         "cluster": {
-            "chief": args.chief_host,
+            "chief": [args.chief_host],
             "worker": args.worker_hosts.split(","),
             "ps": args.ps_hosts.split(",")
         },