Merge pull request #889 from higumachan/patch-1

`example/ssd/train_multi.py` MultiboxTrainChain is not set communicator?
chainer · Jun 6, 2019 · 0942bd8 · 0942bd8
2 parents 9fe62c8 + dbd3348
commit 0942bd8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/ssd/train_multi.py b/examples/ssd/train_multi.py
@@ -84,7 +84,7 @@ def main():
             pretrained_model='imagenet')
 
     model.use_preset('evaluate')
-    train_chain = MultiboxTrainChain(model)
+    train_chain = MultiboxTrainChain(model, comm=comm)
     chainer.cuda.get_device_from_id(device).use()
     model.to_gpu()