From d02b745b81029b2c39c0bb777ff56489ffcc00dc Mon Sep 17 00:00:00 2001 From: JiabinYang Date: Wed, 5 Dec 2018 07:55:35 +0000 Subject: [PATCH] save model per 1000k batches --- fluid/PaddleRec/word2vec/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fluid/PaddleRec/word2vec/train.py b/fluid/PaddleRec/word2vec/train.py index e0a668ef..59c75c3c 100644 --- a/fluid/PaddleRec/word2vec/train.py +++ b/fluid/PaddleRec/word2vec/train.py @@ -203,7 +203,7 @@ def train_loop(args, train_program, reader, py_reader, loss, trainer_id): batch_id) inference_test(global_scope(), model_dir, args) - if batch_id % 100000 == 0 and batch_id != 0: + if batch_id % 1000000 == 0 and batch_id != 0: model_dir = args.model_output_dir + '/batch-' + str( batch_id) fluid.io.save_persistables(executor=exe, dirname=model_dir) -- GitLab