diff --git a/PaddleRec/multi-task/MMoE/README.md b/PaddleRec/multi-task/MMoE/README.md index fb3f23a0c154b7351ea8206bde66f96270d94450..dbe1cc231e490cb25918c15d303508dc214192d7 100644 --- a/PaddleRec/multi-task/MMoE/README.md +++ b/PaddleRec/multi-task/MMoE/README.md @@ -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轮次 ``` 修改脚本的可执行权限并运行 diff --git a/PaddleRec/multi-task/Share_bottom/README.md b/PaddleRec/multi-task/Share_bottom/README.md index 6408e4136850c6c415d7f2cb72ae23d41b2d9e90..83964c288c81701bb6ba68a1231e4f3969ded3ab 100644 --- a/PaddleRec/multi-task/Share_bottom/README.md +++ b/PaddleRec/multi-task/Share_bottom/README.md @@ -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轮次 ``` 修改脚本的可执行权限并运行