提交 c40a0c65 编写于 作者: F frankwhzhang

fix bugs

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