未验证 提交 7c69ddd9 编写于 作者: W WangWenjin 提交者: GitHub

update the description of hyper-parameter gpu_id

上级 3234641a
...@@ -12,7 +12,7 @@ The ogbn-proteins dataset will be downloaded in directory ./dataset automaticall ...@@ -12,7 +12,7 @@ The ogbn-proteins dataset will be downloaded in directory ./dataset automaticall
## How to run ## How to run
```bash ```bash
python train.py --lr 1e-2 --rc 0 --batch_size 1024 --epochs 100 python train.py --lr 1e-2 --rc 0 --batch_size 1024 --epochs 100 --gpu_id 0
``` ```
or or
...@@ -23,6 +23,7 @@ source main.sh ...@@ -23,6 +23,7 @@ source main.sh
### Hyperparameters ### Hyperparameters
- use_gpu: whether to use gpu or not - use_gpu: whether to use gpu or not
- mini_data: use a small dataset to test code - mini_data: use a small dataset to test code
- gpu_id: the id of gpu
- epochs: number of training epochs - epochs: number of training epochs
- lr: learning rate - lr: learning rate
- rc: regularization coefficient - rc: regularization coefficient
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册