Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
haidfs authored Mar 21, 2020
1 parent 7914544 commit 4eb0b03
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,15 +13,14 @@
将TransE类的实例在多进程之间传递
#### Queue版本
./train_fb15k.sh 2
将TransE类的训练数据传入队列,减小进程开销,加快训练速度
当训练完成之后,再进行测试
将TransE类的训练数据传入队列,减小进程开销,加快训练速度
### 注意事项
1. 当训练完成之后,再进行测试。
2. 测试代码需要在Linux环境执行,Windows环境多进程速度慢,且多进程有bug!!!
### 测试部分
#### TestTransEMqQueue
python TestTransEMpQueue.py
多进程队列测试加速,效果不明显,单个测试例0.5s,测试结束需要近5h。
#### TestMainTF
python TestMainTF.py
tf与多进程测试加速,效果显著,测试结束仅需要8min左右。
tf与多进程测试加速,效果显著,Linux环境128G服务器,测试结束仅需要8min左右。
### 最终测试结果
![image](https://github.com/haidfs/TransE/blob/master/images/TestResult.png)
### THANKS TO
Expand Down

0 comments on commit 4eb0b03

Please sign in to comment.