未验证 提交 4f1756ea 编写于 作者: H Huang Zhengjie 提交者: GitHub

Update README.md

上级 a9693b87
...@@ -14,3 +14,5 @@ ogb ...@@ -14,3 +14,5 @@ ogb
``` ```
CUDA_VISIBLE_DEVICES=0,1,2,3 python train.py --use_cuda 1 --num_workers 4 --output_path ./output/model_1 --batch_size 65536 --epoch 1000 --learning_rate 0.005 --hidden_size 256 CUDA_VISIBLE_DEVICES=0,1,2,3 python train.py --use_cuda 1 --num_workers 4 --output_path ./output/model_1 --batch_size 65536 --epoch 1000 --learning_rate 0.005 --hidden_size 256
``` ```
The best record will be saved in ./output/model_1/best.txt.
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册