未验证 提交 8fd5aeda 编写于 作者: W wuzhihua 提交者: GitHub

Merge pull request #231 from 123malin/model_fix

test=develop, bug fix for w2v, youtube_dnn'readme
...@@ -222,15 +222,18 @@ Infer phase2 of epoch 3 done, use time: 4.43099021912, global metrics: acc=[1.] ...@@ -222,15 +222,18 @@ Infer phase2 of epoch 3 done, use time: 4.43099021912, global metrics: acc=[1.]
## 论文复现 ## 论文复现
1. 用原论文的完整数据复现论文效果需要在config.yaml修改超参: 1. 用原论文的完整数据复现论文效果需要在config.yaml修改超参:
```
- name: dataset_train - name: dataset_train
batch_size: 100 # 1. 修改batch_size为100 batch_size: 100 # 1. 修改batch_size为100
type: DataLoader type: DataLoader
data_path: "{workspace}/data/all_train" # 2. 修改数据为全量训练数据 data_path: "{workspace}/data/all_train" # 2. 修改数据为全量训练数据
word_count_dict_path: "{workspace}/data/all_dict/ word_count_dict.txt" # 3. 修改词表为全量词表 word_count_dict_path: "{workspace}/data/all_dict/word_count_dict.txt" # 3. 修改词表为全量词表
data_converter: "{workspace}/w2v_reader.py" data_converter: "{workspace}/w2v_reader.py"
- name: dataset_infer
data_path: "{workspace}/data/all_test" # 4. 修改数据为全量测试数据
word_id_dict_path: "{workspace}/data/all_dict/word_id_dict.txt" # 5. 修改词表为全量词表
- name: single_cpu_train ```
- epochs: # 4. 修改config.yaml中runner的epochs为5。
修改后运行方案:修改config.yaml中的'workspace'为config.yaml的目录位置,执行 修改后运行方案:修改config.yaml中的'workspace'为config.yaml的目录位置,执行
``` ```
......
...@@ -8,7 +8,7 @@ ...@@ -8,7 +8,7 @@
├── data.txt ├── data.txt
├── test ├── test
├── data.txt ├── data.txt
├── generate_ramdom_data # 随机训练数据生成文件 ├── generate_ramdom_data.py # 随机训练数据生成文件
├── __init__.py ├── __init__.py
├── README.md # 文档 ├── README.md # 文档
├── model.py #模型文件 ├── model.py #模型文件
...@@ -107,7 +107,7 @@ python infer.py --use_gpu 1 --test_epoch 19 --inference_model_dir ./inference_yo ...@@ -107,7 +107,7 @@ python infer.py --use_gpu 1 --test_epoch 19 --inference_model_dir ./inference_yo
``` ```
### 运行 ### 运行
``` ```
python -m paddlerec.run -m paddlerec.models.recall.w2v python -m paddlerec.run -m paddlerec.models.recall.youtube_dnn
``` ```
### 结果展示 ### 结果展示
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册