Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
ShomyLiu committed Mar 29, 2021
1 parent bb16c3e commit 29fa183
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,21 @@
A Toolkit for Neural Review-based Recommendation models with Pytorch.
基于评论文本的深度推荐系统模型库 (Pytorch)

**Update: 2021.03.29**

Add a branch (PL) to use [PyTorch Lightning](https://github.com/PyTorchLightning/pytorch-lightning/) to wrap the framework for further distributed training.
```
git clone https://github.com/ShomyLiu/Neu-Review-Rec.git
git checkout pl
```
And the usage:
```
python3 pl_main.py run --use_ddp=True --gpu_id=2
# indicating that using ddp mode for distributed training with 2 gpus. refer `config/config.py`.
```



# Neural Review-based Recommendaton
In this repository, we reimplement some important review-based recommendation models, and provide an extensible framework **NRRec** with Pytorch.
Researchers can implement their own methodss easily in our framework (just in *models* folder).
Expand Down

0 comments on commit 29fa183

Please sign in to comment.