Update model graph visualization notebook (#127)

zhiqwang · Jun 26, 2021 · a7e31a0 · a7e31a0
1 parent 9db63b5
commit a7e31a0
Show file tree

Hide file tree

Showing 4 changed files with 175 additions and 146 deletions.
diff --git a/README.md b/README.md
@@ -109,7 +109,7 @@ We provide a [notebook](notebooks/inference-pytorch-export-libtorch.ipynb) to de
 
 ## 🎨 Model Graph Visualization
 
-Now, `yolort` can draw the model graph directly, checkout our [visualize-jit-models](notebooks/visualize-jit-models.ipynb) notebook to see how to use and visualize the model graph.
+Now, `yolort` can draw the model graph directly, checkout our [model-graph-visualization](notebooks/model-graph-visualization.ipynb) notebook to see how to use and visualize the model graph.
 
 <a href="notebooks/assets/yolov5.detail.svg"><img src="notebooks/assets/yolov5.detail.svg" alt="YOLO model visualize" width="500"/></a>
 

diff --git a/notebooks/README.md b/notebooks/README.md
@@ -1,5 +1,5 @@
 # Notebook Examples
 
-`yolov5rt` can draw the model graph directly, for details see [visualize-jit-models](visualize-jit-models.ipynb) notebook.
+`yolov5rt` can draw the model graph directly, for details see [model-graph-visualization](model-graph-visualization.ipynb) notebook.
 
 <a href="assets/yolov5.detail.svg"><img src="assets/yolov5.detail.svg" alt="YOLO model visualize" width="500"/></a>