Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "24895a1f494062d73028e31880c8848c6a674750"
Commit 2448751c authored by naveen marri's avatar naveen marri Committed by Zihao Ye
Browse files

Fixed the training file (#1109)

parent d16fb3ff
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/pytorch/recommendation/README.md
View file @ 2448751c
...@@ -9,13 +9,13 @@ First, download and extract from https://dgl.ai.s3.us-east-2.amazonaws.com/datas ...@@ -9,13 +9,13 @@ First, download and extract from https://dgl.ai.s3.us-east-2.amazonaws.com/datas
One can then run the following to train PinSage on MovieLens-1M: One can then run the following to train PinSage on MovieLens-1M:
```bash ```bash
python3 train.py --opt Adam --lr 1e-3 --sched none --sgd-switch 25 python3 main.py --opt Adam --lr 1e-3 --sched none --sgd-switch 25
``` ```
One can also incorporate user and movie features into training: One can also incorporate user and movie features into training:
```bash ```bash
python3 train.py --opt Adam --lr 1e-3 --sched none --sgd-switch 25 --use-feature python3 main.py --opt Adam --lr 1e-3 --sched none --sgd-switch 25 --use-feature
``` ```
Currently, performance of PinSage on MovieLens-1M has the best mean reciprocal rank of Currently, performance of PinSage on MovieLens-1M has the best mean reciprocal rank of
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment