baidu / baidu/Dialogue

The matching scores are all equal when training and the model can not converge.

Open
#3 3 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
442
Forks
93
PR merge metrics
No merged PRs in 30d

Description

I am trying to train the model on douban data named "data_small.pkl" .
**My question:** The matching scores are all equal when training and the model can not converge.

The data is downloaded from the link in “ReadMe.txt”.
- tersorflow 1.8
- python 3.5

**1. Here is the dir structrue.**

```
data/douban/
|-- data.pkl
|-- data_small.pkl
|-- dev.txt
|-- test.txt
|-- train.txt
|-- word2id
`-- word_embedding.pkl
```
**2. Here, here's the `main.py` for training.**

```
conf = {
"data_path": "./data/douban/data_small.pkl",
"save_path": "./output/douban/DAM/",
"word_emb_init": "./data/douban/word_embedding.pkl",
"init_model": None, #"./output/douban/DAM_cross/DAM.ckpt", #should be set for test

"rand_seed": None,

"drop_dense": None,
"drop_attention": None,

"is_mask": True,
"is_layer_norm": True,
"is_positional": False,

"stack_num": 5,
"attention_type": "dot",

"learning_rate": 1e-3,
"vocab_size": 172130, #434512
"emb_size": 200,
"batch_size": 256, #200 for test

"max_turn_num": 9,
"max_turn_len": 50,

"max_to_keep": 1,
"num_scan_data": 2,
"_EOS_": 1, #28270
"final_n_class": 1,
}
```

Contributor guide

No contributing guide indexed for this repository

Research direction

Start with the training configuration in main.py and reproduce the reported behavior using data/douban/data_small.pkl with the stated Python 3.5 and TensorFlow 1.8 setup. Inspect the referenced data files and README instructions, then determine why the matching scores remain equal and define completion as training that produces non-equal scores and converges.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, tensorflow
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.