Skip to content

S
seminar-breakout
Commit fd19f4e2 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix a missing chief_only flag

parent f257f0e0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/train/trainers.py
View file @ fd19f4e2
...@@ -385,7 +385,7 @@ class HorovodTrainer(SingleCostTrainer): ...@@ -385,7 +385,7 @@ class HorovodTrainer(SingleCostTrainer):
# the op will be created later in initialize() # the op will be created later in initialize()
self.trainer._broadcast_op.run() self.trainer._broadcast_op.run()
cb = CallbackFactory(trigger=broadcast) cb = CallbackFactory(trigger=broadcast).set_chief_only(False)
return [cb] return [cb]
@HIDE_DOC @HIDE_DOC
......
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