Update README.md

Fangyh09 · Jan 4, 2024 · 3989601 · 3989601
1 parent ced34c6
commit 3989601
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/README.md b/README.md
@@ -7,6 +7,7 @@ Compute CNN receptive field size in pytorch. Support both 2D CNN and 3D CNN.
 
 ## Create 2D visualization gif
 ```python
+# pip install git+ssh://git@github.com/Fangyh09/pytorch-receptive-field.git
 from torch_receptive_field import receptive_field
 from torch_receptive_field import receptive_field_visualization_2d