提交 f06b64fe 编写于 作者: Y Yu Yang

Test GPU

上级 5bca268b
...@@ -54,7 +54,7 @@ def input_order_converter(generator): ...@@ -54,7 +54,7 @@ def input_order_converter(generator):
def main(): 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( config = paddle.trainer.config_parser.parse_config(
'simple_mnist_network.py', '') 'simple_mnist_network.py', '')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册