Skip to content

S
seminar-breakout
You need to sign in or sign up before continuing.
Commit 2be64ce0 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix missing __init__ (#505)

parent 4e933ef9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tensorpack/train/trainers.py
View file @ 2be64ce0
...@@ -157,6 +157,7 @@ class DistributedTrainerReplicated(SingleCostTrainer): ...@@ -157,6 +157,7 @@ class DistributedTrainerReplicated(SingleCostTrainer):
else: else:
self.is_chief = False self.is_chief = False
logger.info("Distributed training on cluster:\n" + str(server.server_def.cluster)) logger.info("Distributed training on cluster:\n" + str(server.server_def.cluster))
super(DistributedTrainerReplicated, self).__init__()
def _setup_input(self, inputs_desc, input): def _setup_input(self, inputs_desc, input):
if self.job_name == 'ps': if self.job_name == 'ps':
......
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