提交 eb5cf64b 编写于 作者: L liaogang

reduce data_layer size of unit test to avoid cuda out of memory on MAC OS

上级 d3eef0c9
...@@ -16,9 +16,9 @@ ...@@ -16,9 +16,9 @@
from paddle.trainer_config_helpers import * from paddle.trainer_config_helpers import *
settings(batch_size=1000) settings(batch_size=300)
data = data_layer(name ="input", size=100000) data = data_layer(name ="input", size=10000)
proj1 = table_projection(input=data, size=128) proj1 = table_projection(input=data, size=128)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册