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()