Skip to content

Commit

Permalink
Update docs/source/api/notebooks/Notebook_6_API2CLI.ipynb
Browse files Browse the repository at this point in the history
Co-authored-by: xiang song(charlie.song) <[email protected]>
  • Loading branch information
zhjwy9343 and classicsong authored Sep 30, 2024
1 parent d01e6c6 commit 87ea784
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/source/api/notebooks/Notebook_6_API2CLI.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
"source": [
"## Brief Introduction and Run CLIs on a Single Machine\n",
"\n",
"In order to use GraphStorm CLIs, we need to put the customized model into a Python file, which can be called in the [Task-agnostic CLI for model training and inference](https://graphstorm.readthedocs.io/en/latest/cli/model-training-inference/single-machine-training-inference.html#task-agnostic-cli-for-model-training-and-inference) as an argument. We could build two files for model training and inference separately.\n",
"In order to use GraphStorm CLIs, we need to put the custom model into a Python file, which can be called in the [Task-agnostic CLI for model training and inference](https://graphstorm.readthedocs.io/en/latest/cli/model-training-inference/single-machine-training-inference.html#task-agnostic-cli-for-model-training-and-inference) as an argument. We build two files for model training and inference separately.\n",
"\n",
"We can reuse most of the code about the customized `RGAT` module in Notebook 4 in the training and inference files, i.e., copy and paste the code of class `Ara_GatLayer`, `Ara_GatEncoder`, and `RgatNCModel` into the two files. \n",
"\n",
Expand Down

0 comments on commit 87ea784

Please sign in to comment.