Test GPU

f06b64fe · Yu Yang · 5bca268b · f06b64fe
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

demo/mnist/api_train.py demo/mnist/api_train.py +1 -1

未找到文件。
--- a/demo/mnist/api_train.py
+++ b/demo/mnist/api_train.py
@@ -54,7 +54,7 @@ def input_order_converter(generator):


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