README.md 364 字节
Newer Older
Z
zhang wenhui 已提交
1
# gru4rec 动态图实现
Z
zhang wenhui 已提交
2

Z
zhang wenhui 已提交
3 4 5 6 7
# 环境配置
paddle 1.7



Z
zhang wenhui 已提交
8 9 10 11 12 13
# 下载数据
```
wget https://paddlerec.bj.bcebos.com/gru4rec/dy_graph/data_rsc15.tar
tar xvf data_rsc15.tar
```

Z
zhang wenhui 已提交
14 15 16
# 数据格式
数据格式及预处理处理同静态图相同。

Z
zhang wenhui 已提交
17 18 19 20 21 22 23
# 训练及预测

```
CUDA_VISIBLE_DEVICES=0 nohup sh run_gru.sh > log 2>&1 &
```

每一轮训练完都会进行预测。