fix

Author: yuyu2172

examples/pspnet/train_multi.py

@@ -207,7 +207,7 @@ def main():
             n_class, pretrained_model='imagenet',
             input_size=dataset_cfg['input_size'])
     train_chain = TrainChain(model)
-    create_mnbn_model(model, comm)
+    model = create_mnbn_model(model, comm)
     if device >= 0:
         chainer.cuda.get_device_from_id(device).use()
         train_chain.to_gpu()