File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -33,14 +33,14 @@ git clone https://github.com/PaddlePaddle/GraphNet.git
3333cd 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
3939python 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/ )
4242python -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 = ...
5555model = 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
6161For details, see docstring of ` graph_net.torch.extract ` defined in ` graph_net/torch/extractor.py `
You can’t perform that action at this time.
0 commit comments