提交 9ba2c951 编写于 作者: B barrierye

fix code and readme

上级 424ba529
...@@ -12,12 +12,20 @@ FL-mobile是一个集移动端算法模拟调研、训练和部署为一体的 ...@@ -12,12 +12,20 @@ FL-mobile是一个集移动端算法模拟调研、训练和部署为一体的
## 准备工作 ## 准备工作
- 安装mpirun - 安装mpirun
- python安装grpc - python安装grpc
```shell ```shell
pip install grpcio==1.28.1 pip install grpcio==1.28.1
``` ```
- 安装Paddle - 安装Paddle
```shell
pip install paddlepaddle==1.8.0
```
## 快速开始 ## 快速开始
我们以Leaf数据集中的[reddit数据](https://github.com/TalwalkarLab/leaf/tree/master/data/reddit)为例,用LSTM建模,在simulator 我们以Leaf数据集中的[reddit数据](https://github.com/TalwalkarLab/leaf/tree/master/data/reddit)为例,用LSTM建模,在simulator
...@@ -89,7 +97,7 @@ framework.py : INFO infer results: 0.116334 ...@@ -89,7 +97,7 @@ framework.py : INFO infer results: 0.116334
- Step1 模型初始化 - Step1 模型初始化
1. 全局参数初始化:由编号为0的simulator来做模型初始化工作,初始化之后,它会通过UpdateGlobalParams()接口将参数传递给Scheduler; 1. 全局参数初始化:由编号为0的simulator来做模型初始化工作,初始化之后,它会通过UpdateGlobalParams()接口将参数传递给Scheduler;
2. 个性化参数初始化 2. 个性化参数初始化
- Step2 模型分发 - Step2 模型分发
......
...@@ -194,7 +194,8 @@ class LanguageModel(ModelBase): ...@@ -194,7 +194,8 @@ class LanguageModel(ModelBase):
- fluid.layers.reduce_sum(correct_pad) - fluid.layers.reduce_sum(correct_pad)
self.loss_ = loss self.loss_ = loss
self.correct_ = eval_metric_opsself.input_name_list_ = [ self.correct_ = eval_metric_ops
self.input_name_list_ = [
'features', 'labels', 'seq_len_ph', 'seq_mask_ph', 'init_hidden', 'features', 'labels', 'seq_len_ph', 'seq_mask_ph', 'init_hidden',
'init_cell' 'init_cell'
] ]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册