From dbd3348ccda5768fd0599272d1645121900a4576 Mon Sep 17 00:00:00 2001 From: Yuta Hinokuma Date: Wed, 5 Jun 2019 17:51:54 +0900 Subject: [PATCH] `example/ssd/train_multi.py` MultiboxTrainChain is not set communicator? --- examples/ssd/train_multi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/ssd/train_multi.py b/examples/ssd/train_multi.py index 61f51d9ec5..127081a01d 100644 --- 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()