Skip to content

Commit f06b64f

Browse files
committed
Test GPU
1 parent 5bca268 commit f06b64f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

demo/mnist/api_train.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def input_order_converter(generator):
5454

5555

5656
def main():
57-
api.initPaddle("-use_gpu=false", "-trainer_count=4") # use 4 cpu cores
57+
api.initPaddle("-use_gpu=true", "-trainer_count=4") # use 4 cpu cores
5858
config = paddle.trainer.config_parser.parse_config(
5959
'simple_mnist_network.py', '')
6060

0 commit comments

Comments
 (0)