[Example] Add TorchMetrics in README (dmlc#3913)

* Update README.md * Update README.md * Update README.md Co-authored-by: Rhett Ying <[email protected]>
anirband · Apr 9, 2022 · 01e5062 · 01e5062
1 parent adb3a7c
commit 01e5062
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/examples/pytorch/cluster_gcn/README.md b/examples/pytorch/cluster_gcn/README.md
@@ -10,7 +10,7 @@ Dependencies
 - Python 3.7+(for string formatting features)
 - PyTorch 1.9.0+
 - sklearn
-
+- TorchMetrics
 
 ## Run Experiments
 

diff --git a/examples/pytorch/graphsage/README.md b/examples/pytorch/graphsage/README.md
@@ -9,13 +9,11 @@ Advanced usages, including how to run pure GPU sampling, how to train with PyTor
 
 Requirements
 ------------
-- requests
 
 ```bash
-pip install requests
+pip install requests torchmetrics
 ```
 
-
 Results
 -------
 

diff --git a/examples/pytorch/rgat/README.md b/examples/pytorch/rgat/README.md
@@ -1,6 +1,9 @@
 Relational GAT
 ==============
 
+Requirements:
+- TorchMetrics
+
 This is an adaptation of RGCN where the graph convolution is replaced with graph attention.
 
 Run