提交 42143e86 编写于 作者: Q qiaolongfei

add paddle.init and import to support ipynb

上级 e1b39252
...@@ -216,6 +216,16 @@ cd data ...@@ -216,6 +216,16 @@ cd data
## 训练流程说明 ## 训练流程说明
### paddle初始化
```python
# 加载 paddle的python包
import paddle.v2 as paddle
# 配置只使用cpu,并且使用一个cpu进行训练
paddle.init(use_gpu=False, trainer_count=1)
```
### 数据定义 ### 数据定义
首先要定义词典大小,数据生成和网络配置都需要用到。然后获取wmt14的dataset reader。 首先要定义词典大小,数据生成和网络配置都需要用到。然后获取wmt14的dataset reader。
......
...@@ -110,11 +110,12 @@ def main(): ...@@ -110,11 +110,12 @@ def main():
update_equation=optimizer) update_equation=optimizer)
# define data reader # define data reader
reader_dict = { feeding = {
'source_language_word': 0, 'source_language_word': 0,
'target_language_word': 1, 'target_language_word': 1,
'target_language_next_word': 2 'target_language_next_word': 2
} }
wmt14_reader = paddle.batch( wmt14_reader = paddle.batch(
paddle.reader.shuffle( paddle.reader.shuffle(
paddle.dataset.wmt14.train(dict_size=dict_size), buf_size=8192), paddle.dataset.wmt14.train(dict_size=dict_size), buf_size=8192),
...@@ -132,7 +133,7 @@ def main(): ...@@ -132,7 +133,7 @@ def main():
reader=wmt14_reader, reader=wmt14_reader,
event_handler=event_handler, event_handler=event_handler,
num_passes=10000, num_passes=10000,
reader_dict=reader_dict) feeding=feeding)
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册