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

9 years ago · eb5cf64b5e
parent d3eef0c9af
commit eb5cf64b5e
1 changed files with 2 additions and 2 deletions
--- a/paddle/gserver/tests/concat_table_b.conf
+++ b/paddle/gserver/tests/concat_table_b.conf
@ -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)

 proj1 = table_projection(input=data, size=128)