Skip to content

Commit

Permalink
Add GNN documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
davidjurado committed Sep 3, 2024
1 parent adb45e5 commit 4557eb4
Show file tree
Hide file tree
Showing 4 changed files with 53 additions and 1 deletion.
2 changes: 1 addition & 1 deletion docs/minified-benchmarks/bert.md
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ In the Google Cloud console, search for the Cloud TPU API page, then click Enabl

Then go to the virtual machine sections and select [TPUs](https://console.cloud.google.com/compute/tpus)

Select create TPU node, fill in all the needed parameters, the recommended TPU type in the [readme](../README.md#on-tpu-v3-128) is v3-128 and the recommended TPU software version is 2.4.0.
Select create TPU node, fill in all the needed parameters, the recommended TPU type in the [readme](https://github.com/mlcommons/training/blob/3283fc35e68deb88f7197155964f7c3858705649/language_model/tensorflow/bert/README.md#on-tpu-v3-128) is v3-128 and the recommended TPU software version is 2.4.0.

The 3 most important parameters you need to remember are: `project name`, `TPU name`, and `TPU Zone`.

Expand Down
50 changes: 50 additions & 0 deletions docs/minified-benchmarks/gnn.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
# Graph Neural Network

The benchmark reference for Graph Neural Network can be found in this [link](https://github.com/mlcommons/training/tree/master/graph_neural_network), and here is the PR for the minified benchmark implementation: [link](https://github.com/mlcommons/training/pull/762).

## Project setup

An important requirement is that you must have Docker installed.

```bash
# Create Python environment and install MLCube Docker runner
virtualenv -p python3 ./env && source ./env/bin/activate && pip install pip==24.0 && pip install mlcube-docker
# Fetch the implementation from GitHub
git clone https://github.com/mlcommons/training && cd ./training
git fetch origin pull/762/head:feature/mlcube_graph_nn && git checkout feature/mlcube_graph_nn
cd ./graph_neural_network/mlcube
```

Inside the mlcube directory run the following command to check implemented tasks.

```shell
mlcube describe
```

### MLCube tasks

Download dataset.

```shell
mlcube run --task=download_data -Pdocker.build_strategy=always
```

Process dataset.

```shell
mlcube run --task=process_data -Pdocker.build_strategy=always
```

Train GNN.

```shell
mlcube run --task=train -Pdocker.build_strategy=always
```

### Execute the complete pipeline

You can execute the complete pipeline with one single command.

```shell
mlcube run --task=download_data,process_data,train -Pdocker.build_strategy=always
```
1 change: 1 addition & 0 deletions docs/minified-benchmarks/introduction.md
Original file line number Diff line number Diff line change
Expand Up @@ -18,3 +18,4 @@ The main advantages of these minified benchmarks are:
- [ResNet](resnet.md)
- [Bert](bert.md)
- [Object Detection](object-detection.md)
- [Graph Neural Network](gnn.md)
1 change: 1 addition & 0 deletions mkdocs.yml
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@ nav:
- ResNet: minified-benchmarks/resnet.md
- Bert: minified-benchmarks/bert.md
- Object Detection: minified-benchmarks/object-detection.md
- Graph Neural Network: minified-benchmarks/gnn.md

theme:
features:
Expand Down

0 comments on commit 4557eb4

Please sign in to comment.