提交 acdfd00f 编写于 作者: B barrierye

update README

上级 9ba2c951
...@@ -28,8 +28,7 @@ FL-mobile是一个集移动端算法模拟调研、训练和部署为一体的 ...@@ -28,8 +28,7 @@ FL-mobile是一个集移动端算法模拟调研、训练和部署为一体的
## 快速开始 ## 快速开始
我们以Leaf数据集中的[reddit数据](https://github.com/TalwalkarLab/leaf/tree/master/data/reddit)为例,用LSTM建模,在simulator 我们以Leaf数据集中的[reddit数据](https://github.com/TalwalkarLab/leaf/tree/master/data/reddit)为例,参考[这篇论文](https://arxiv.org/pdf/1812.01097.pdf),用LSTM建模,在simulator中给出一个单机训练的例子。通过这个例子,您能了解simulator的基础用法。
中给出一个单机训练的例子,通过这个例子,您能了解simulator的基础用法。
### 准备数据 ### 准备数据
...@@ -60,6 +59,8 @@ cd .. ...@@ -60,6 +59,8 @@ cd ..
### 开始训练 ### 开始训练
在训练中,我们每轮用均匀采样(Uniform Sample)方式选取`10`个Client进行训练,每个Client在本地用该Client对应的全部数据(未经shuffle)训练`1`个epoch,总共训练`100`轮。在本实验中使用的Client学习率为`1.0`,FedAvg学习率为`1.85 `
```shell ```shell
export PYTHONPATH=$PWD:$PYTHONPATH export PYTHONPATH=$PWD:$PYTHONPATH
mpirun -np 2 python application.py lm_data mpirun -np 2 python application.py lm_data
...@@ -67,11 +68,13 @@ mpirun -np 2 python application.py lm_data ...@@ -67,11 +68,13 @@ mpirun -np 2 python application.py lm_data
### 训练结果 ### 训练结果
在测试集上,测试Top1为 `11.6% `
```shell ```shell
framework.py : INFO infer results: 0.116334 framework.py : INFO infer results: 0.116334
``` ```
即:在测试集上的,测试Top1为 11.6% 相同参数的非联邦训练测试Top1为`11.1%`
## 添加自己的数据集和Trainer ## 添加自己的数据集和Trainer
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册