File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
python/paddle/fluid/tests/book Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change 38
38
mix_hidden_lr = 1e-3
39
39
40
40
IS_SPARSE = True
41
- PASS_NUM = 1
41
+ PASS_NUM = 2
42
42
BATCH_SIZE = 10
43
43
44
44
embedding_name = 'emb'
@@ -196,7 +196,7 @@ def train_loop(main_program):
196
196
print ("second per batch: " + str ((time .time (
197
197
) - start_time ) / batch_id ))
198
198
# Set the threshold low to speed up the CI test
199
- if float (cost ) < 60 .0 :
199
+ if float (cost ) < 80 .0 :
200
200
if save_dirname is not None :
201
201
# TODO(liuyiqun): Change the target to crf_decode
202
202
fluid .io .save_inference_model (save_dirname , [
@@ -208,6 +208,10 @@ def train_loop(main_program):
208
208
209
209
batch_id = batch_id + 1
210
210
211
+ raise RuntimeError (
212
+ "This model should save_inference_model and return, but not reach here, please check!"
213
+ )
214
+
211
215
if is_local :
212
216
train_loop (fluid .default_main_program ())
213
217
else :
You can’t perform that action at this time.
0 commit comments