diff --git a/examples/gin/README.md b/examples/gin/README.md index 22145cb04236c2522e7e1d6be8e5b8b6e1bd913a..e91b6acb70ff3f320f0ec648279d25d7eb2c1660 100644 --- a/examples/gin/README.md +++ b/examples/gin/README.md @@ -4,7 +4,8 @@ ### Datasets -The dataset can be downloaded from [here](https://github.com/weihua916/powerful-gnns/blob/master/dataset.zip) +The dataset can be downloaded from [here](https://github.com/weihua916/powerful-gnns/blob/master/dataset.zip). +After downloading the data,uncompress them, then a directory named `./dataset/` can be found in current directory. Note that the current directory is the root directory of GIN model. ### Dependencies @@ -15,7 +16,7 @@ The dataset can be downloaded from [here](https://github.com/weihua916/powerful- For examples, use GPU to train GIN model on MUTAG dataset. ``` -python main.py --use_cuda --dataset_name MUTAG +python main.py --use_cuda --dataset_name MUTAG --data_path ./dataset ``` ### Hyperparameters