提交 d3eef0c9 编写于 作者: L liaogang

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

上级 9a9e0597
......@@ -16,9 +16,9 @@
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)
# emb1 is equal to emb2, note that bias_attr=false
# and act=LinearActivation() in default.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册