Commit 47bddd8e authored by Yuxin Wu's avatar Yuxin Wu

fix distributed trainer docs formatting (fix #375)

parent 6d954998
...@@ -62,7 +62,7 @@ class DistributedTrainerReplicated(MultiGPUTrainerBase): ...@@ -62,7 +62,7 @@ class DistributedTrainerReplicated(MultiGPUTrainerBase):
config=get_default_sess_config()) config=get_default_sess_config())
DistributedTrainerReplicated(config, server).train() DistributedTrainerReplicated(config, server).train()
.. code-block:: .. code-block:: none
# start your jobs: # start your jobs:
(host1)$ train.py --job worker --task 0 (host1)$ train.py --job worker --task 0
...@@ -74,7 +74,7 @@ class DistributedTrainerReplicated(MultiGPUTrainerBase): ...@@ -74,7 +74,7 @@ class DistributedTrainerReplicated(MultiGPUTrainerBase):
""" """
Args: Args:
config(TrainConfig): Must contain 'model' and 'data'. config(TrainConfig): Must contain 'model' and 'data'.
server (tf.train.Server): the server object with ps and workers server(tf.train.Server): the server object with ps and workers
""" """
assert config.data is not None and config.model is not None assert config.data is not None and config.model is not None
self.server = server self.server = server
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment