Skip to content

Commit f4625d5

Browse files
committedJan 27, 2022
update docs
1 parent ba2c820 commit f4625d5

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed
 

‎gnnrl/graph_env/graph_construction.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,6 @@ def net_info(model_name):
154154
in_channels.append(layer.in_channels)
155155
out_channels.append(layer.out_channels)
156156
return in_channels,out_channels,[]
157-
158157
elif model_name == 'resnet50':
159158
in_channels = []
160159
out_channels=[]
@@ -166,6 +165,7 @@ def net_info(model_name):
166165
in_channels.append(layer.in_channels)
167166
out_channels.append(layer.out_channels)
168167
return in_channels,out_channels,[]
168+
169169
def create_edge_features(edge_types,type_features,device):
170170
if max(edge_types)> len(type_features):
171171
#random initial primitive operation like batch norm

‎mkdocs.yml

+2-1
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,8 @@ nav:
2727
- Rewrite the Graph Environment: empty.md
2828

2929
- About Us:
30-
- SwAPP: swapp.md
30+
# - SwAPP: swapp.md
31+
- SwAPP: empty.md
3132
- API:
3233
- gnnrl.lib: empty.md
3334
- gnnrl.graph_env: empty.md

0 commit comments

Comments
 (0)
Please sign in to comment.