Skip to content

Commit e4d8f47

Browse files
committed
change the target cost of test_label_semantic_roles to speed up test
1 parent 2354409 commit e4d8f47

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/paddle/fluid/tests/book/test_label_semantic_roles.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
mix_hidden_lr = 1e-3
3939

4040
IS_SPARSE = True
41-
PASS_NUM = 1
41+
PASS_NUM = 2
4242
BATCH_SIZE = 10
4343

4444
embedding_name = 'emb'
@@ -196,7 +196,7 @@ def train_loop(main_program):
196196
print("second per batch: " + str((time.time(
197197
) - start_time) / batch_id))
198198
# Set the threshold low to speed up the CI test
199-
if float(cost) < 60.0:
199+
if float(cost) < 80.0:
200200
if save_dirname is not None:
201201
# TODO(liuyiqun): Change the target to crf_decode
202202
fluid.io.save_inference_model(save_dirname, [

0 commit comments

Comments
 (0)