Skip to content

Commit fa2ed23

Browse files
authored
Update README.md
1 parent 8c4147b commit fa2ed23

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,14 +33,14 @@ git clone https://github.com/PaddlePaddle/GraphNet.git
3333
cd GraphNet
3434

3535
# Set your workspace directory
36-
export GRAPH_NET_EXTRACT_WORKSPACE=/home/yourname/graphnet_workspace
36+
export GRAPH_NET_EXTRACT_WORKSPACE=/home/yourname/graphnet_workspace/
3737

3838
# Extract the ResNet‑18 computation graph
3939
python graph_net/test/vision_model_test.py
4040

41-
# Validate the extracted graph (e.g. /home/yourname/graphnet_workspace/resnet18)
41+
# Validate the extracted graph (e.g. /home/yourname/graphnet_workspace/resnet18/)
4242
python -m graph_net.torch.validate \
43-
--model-path $GRAPH_NET_EXTRACT_WORKSPACE/resnet18
43+
--model-path $GRAPH_NET_EXTRACT_WORKSPACE/resnet18/
4444
```
4545

4646
**Step 1: graph_net.torch.extract**
@@ -55,7 +55,7 @@ model = ...
5555
model = graph_net.torch.extract(name="model_name")(model)
5656

5757
# After running, the extracted graph will be saved to:
58-
# $GRAPH_NET_EXTRACT_WORKSPACE/model_name
58+
# $GRAPH_NET_EXTRACT_WORKSPACE/model_name/
5959
```
6060

6161
For details, see docstring of `graph_net.torch.extract` defined in `graph_net/torch/extractor.py`

0 commit comments

Comments
 (0)