Update README.md

Deep-Learning-Profiling-Tools · Apr 21, 2024 · d708e38 · d708e38
1 parent d076e3e
commit d708e38
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -46,9 +46,9 @@ python rgat.py --device cuda --mode fasten --dataset AIFB
 cd examples/rgcn
 # Without fasten
 # Available datasets are: DBLP, Freebase, AIFB, MUTAG, BGS, AM
-python rgcn.py --device cuda --dataset DBLP
+python rgcn.py --device cuda --example DBLP
 # With fasten
-python rgat.py --device cuda --mode fasten --dataset DBLP
+python rgat.py --device cuda --mode fasten --example DBLP
 ```
 
 - RGAT