fix assertion

chainer · Dec 19, 2018 · b276218 · b276218
1 parent c34832d
commit b276218
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/semantic_segmentation/eval_semantic_segmentation_multi.py b/examples/semantic_segmentation/eval_semantic_segmentation_multi.py
@@ -33,9 +33,9 @@ def main():
     dataset, label_names, model = get_dataset_and_model(
         args.dataset, args.model, args.pretrained_model,
         (args.input_size, args.input_size))
-    assert (len(dataset) % comm.size == 0, \
-            "The size of the dataset should be a multiple "\
-            "of the number of GPUs")
+    assert len(dataset) % comm.size == 0, \
+        "The size of the dataset should be a multiple "\
+        "of the number of GPUs"
 
     chainer.cuda.get_device_from_id(device).use()
     model.to_gpu()