Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
arjunsuresh authored Dec 20, 2024
1 parent 45bb01b commit 293d4ea
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion graph/R-GAT/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,8 @@ This is the reference implementation for MLPerf Inference Graph Neural Network.
## Automated command to run the benchmark via MLCommons CM

Please see the [new docs site](https://docs.mlcommons.org/inference/benchmarks/graph/rgat/) for an automated way to run this benchmark across different available implementations and do an end-to-end submission with or without docker.

You can also do `pip install cm4mlops` and then use `cm` commands for downloading the model and datasets using the commands given in the later sections.

## Setup
Set the following helper variables
Expand Down Expand Up @@ -78,7 +80,6 @@ pip install dgl -f https://data.dgl.ai/wheels/torch-2.1/repo.html
### Download model through CM (Collective Minds)

```
pip install cm4mlops
cm run script --tags=get,ml-model,rgat --outdirname=<path_to_download>
```

Expand Down

0 comments on commit 293d4ea

Please sign in to comment.