fix style

PaddlePaddle · Feb 1, 2018 · 52df85f · 52df85f
1 parent 978396e
commit 52df85f
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/benchmark/cluster/vgg16/vgg16_fluid.py b/benchmark/cluster/vgg16/vgg16_fluid.py
@@ -50,11 +50,7 @@ def str2bool(v):
     default='CPU',
     choices=['CPU', 'GPU'],
     help="The device type.")
-parser.add_argument(
-    '--device_id',
-    type=int,
-    default=0,
-    help="The device id.")
+parser.add_argument('--device_id', type=int, default=0, help="The device id.")
 parser.add_argument(
     '--data_format',
     type=str,
@@ -140,7 +136,8 @@ def main():
     optimize_ops, params_grads = optimizer.minimize(avg_cost)
 
     # Initialize executor
-    place = core.CPUPlace() if args.device == 'CPU' else core.CUDAPlace(args.device_id)
+    place = core.CPUPlace() if args.device == 'CPU' else core.CUDAPlace(
+        args.device_id)
     exe = fluid.Executor(place)
 
     # test