example/ssd/train_multi.py MultiboxTrainChain is not set communicator?

chainer · Jun 5, 2019 · dbd3348 · dbd3348
1 parent 9fe62c8
commit dbd3348
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()