diff --git a/demo/word2vec/api_train_v2.py b/demo/word2vec/api_train_v2.py index 8fea6feeef0c4de49e10e716769200f5e22d05ff..98ade830cf51e69e6b9ff4097090219c100f0cdf 100644 --- a/demo/word2vec/api_train_v2.py +++ b/demo/word2vec/api_train_v2.py @@ -69,7 +69,8 @@ def main(): def event_handler(event): if isinstance(event, paddle.event.EndIteration): if event.batch_id % 100 == 0: - trainer.save_parameter_to_tar("output", "batch-" + str(event.batch_id)) + trainer.save_parameter_to_tar("output", + "batch-" + str(event.batch_id)) result = trainer.test( paddle.batch( paddle.dataset.imikolov.test(word_dict, N), 32))