提交 e88a3a25 编写于 作者: O overlordmax

edit README.md

上级 145cf99b
......@@ -59,15 +59,15 @@ GPU环境
```sh
python train_mmoe.py --use_gpu True \ #使用gpu训练
--train_path data/data24913/train_data/\ #训练数据路径
--test_path data/data24913/test_data/\ #测试数据路径
--feature_size 499\ #设置特征的维度
--batch_size 32\ #设置batch_size大小
--expert_num 8\ #设置expert数量
--gate_num 2\ #设置gate数量
--expert_size 16\ #设置expert网络大小
--tower_size 8\ #设置tower网络大小
--epochs 400 #设置epoch轮次
--train_path data/data24913/train_data/\ #训练数据路径
--test_path data/data24913/test_data/\ #测试数据路径
--feature_size 499\ #设置特征的维度
--batch_size 32\ #设置batch_size大小
--expert_num 8\ #设置expert数量
--gate_num 2\ #设置gate数量
--expert_size 16\ #设置expert网络大小
--tower_size 8\ #设置tower网络大小
--epochs 400 #设置epoch轮次
```
修改脚本的可执行权限并运行
......@@ -82,15 +82,15 @@ CPU环境
```sh
python train_mmoe.py --use_gpu False \ #使用cpu训练
--train_path data/data24913/train_data/\ #训练数据路径
--test_path data/data24913/test_data/\ #测试数据路径
--feature_size 499\ #设置特征的维度
--batch_size 32\ #设置batch_size大小
--expert_num 8\ #设置expert数量
--gate_num 2\ #设置gate数量
--expert_size 16\ #设置expert网络大小
--tower_size 8\ #设置tower网络大小
--epochs 400 #设置epoch轮次
--train_path data/data24913/train_data/\ #训练数据路径
--test_path data/data24913/test_data/\ #测试数据路径
--feature_size 499\ #设置特征的维度
--batch_size 32\ #设置batch_size大小
--expert_num 8\ #设置expert数量
--gate_num 2\ #设置gate数量
--expert_size 16\ #设置expert网络大小
--tower_size 8\ #设置tower网络大小
--epochs 400 #设置epoch轮次
```
修改脚本的可执行权限并运行
......
......@@ -59,14 +59,14 @@ GPU环境
```sh
python share_bottom.py --use_gpu True\ #使用gpu训练
--train_path data/data24913/train_data/\ #训练数据路径
--test_path data/data24913/test_data/\ #测试数据路径
--batch_size 32\ #设置batch_size大小
--feature_size 499\ #设置特征维度
--bottom_size 117\ #设置bottom网络大小
--tower_nums 2\ #设置tower数量
--tower_size 8\ #设置tower网络大小
--epochs 400 #设置epoch轮次
--train_path data/data24913/train_data/\ #训练数据路径
--test_path data/data24913/test_data/\ #测试数据路径
--batch_size 32\ #设置batch_size大小
--feature_size 499\ #设置特征维度
--bottom_size 117\ #设置bottom网络大小
--tower_nums 2\ #设置tower数量
--tower_size 8\ #设置tower网络大小
--epochs 400 #设置epoch轮次
```
修改脚本的可执行权限并运行
......@@ -81,14 +81,14 @@ CPU环境
```sh
python share_bottom.py --use_gpu False\ #使用cpu训练
--train_path data/data24913/train_data/\ #训练数据路径
--test_path data/data24913/test_data/\ #测试数据路径
--batch_size 32\ #设置batch_size大小
--feature_size 499\ #设置特征维度
--bottom_size 117\ #设置bottom网络大小
--tower_nums 2\ #设置tower数量
--tower_size 8\ #设置tower网络大小
--epochs 400 #设置epoch轮次
--train_path data/data24913/train_data/\ #训练数据路径
--test_path data/data24913/test_data/\ #测试数据路径
--batch_size 32\ #设置batch_size大小
--feature_size 499\ #设置特征维度
--bottom_size 117\ #设置bottom网络大小
--tower_nums 2\ #设置tower数量
--tower_size 8\ #设置tower网络大小
--epochs 400 #设置epoch轮次
```
修改脚本的可执行权限并运行
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册