提交 e88a3a25 编写于 作者: O overlordmax

edit README.md

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