提交 c40a0c65 编写于 作者: F frankwhzhang

fix bugs

上级 0683deac
......@@ -69,7 +69,6 @@ sh run.sh
### 训练
```
cd modles/multitask/mmoe # 进入选定好的排序模型的目录 以MMoE为例
python -m paddlerec.run -m paddlerec.models.multitask.mmoe # 使用内置配置
python -m paddlerec.run -m ./config.yaml # 自定义修改超参后,指定配置文件,使用自定义配置
```
......
......@@ -30,7 +30,7 @@ hyper_parameters:
num_users: 6040
num_items: 3706
latent_dim: 8
layers: [64, 32, 16, 8]
fc_layers: [64, 32, 16, 8]
optimizer:
class: adam
learning_rate: 0.001
......
......@@ -28,7 +28,7 @@ class Model(ModelBase):
self.num_users = envs.get_global_env("hyper_parameters.num_users")
self.num_items = envs.get_global_env("hyper_parameters.num_items")
self.latent_dim = envs.get_global_env("hyper_parameters.latent_dim")
self.layers = envs.get_global_env("hyper_parameters.layers")
self.layers = envs.get_global_env("hyper_parameters.fc_layers")
def input_data(self, is_infer=False, **kwargs):
user_input = fluid.data(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册