From ca3aa4153559a82b79bd155ec2a596db4eaf9695 Mon Sep 17 00:00:00 2001 From: Li Fuchen Date: Thu, 20 Feb 2020 09:43:40 +0800 Subject: [PATCH] update ml API to 1.7 (#4314) --- PaddleNLP/language_model/README.md | 2 +- PaddleNLP/language_model/train.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/PaddleNLP/language_model/README.md b/PaddleNLP/language_model/README.md index 10b882a7..8bf919a3 100644 --- a/PaddleNLP/language_model/README.md +++ b/PaddleNLP/language_model/README.md @@ -5,7 +5,7 @@ ## 1. 任务说明 本文主要介绍基于lstm的语言的模型的实现,给定一个输入词序列(中文分词、英文tokenize),计算其ppl(语言模型困惑度,用户表示句子的流利程度),基于循环神经网络语言模型的介绍可以[参阅论文](https://arxiv.org/abs/1409.2329)。相对于传统的方法,基于循环神经网络的方法能够更好的解决稀疏词的问题。 -**目前语言模型要求使用PaddlePaddle 1.6及以上版本或适当的develop版本。** +**目前语言模型要求使用PaddlePaddle 1.7及以上版本或适当的develop版本。** 同时推荐用户参考[IPython Notebook demo](https://aistudio.baidu.com/aistudio/projectDetail/122290) diff --git a/PaddleNLP/language_model/train.py b/PaddleNLP/language_model/train.py index f00604f3..5a6773c2 100644 --- a/PaddleNLP/language_model/train.py +++ b/PaddleNLP/language_model/train.py @@ -60,7 +60,7 @@ def profile_context(profile=True, profiler_path='/tmp/paddingrnn.profile'): def get_current_model_para(train_prog, train_exe): - param_list = train_prog.block(0).all_parameters() + param_list = train_prog.all_parameters() param_name_list = [p.name for p in param_list] vals = {} @@ -73,7 +73,7 @@ def get_current_model_para(train_prog, train_exe): def save_para_npz(train_prog, train_exe): print("begin to save model to model_base") - param_list = train_prog.block(0).all_parameters() + param_list = train_prog.all_parameters() param_name_list = [p.name for p in param_list] vals = {} -- GitLab