File tree 2 files changed +3
-2
lines changed
2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -154,7 +154,6 @@ def net_info(model_name):
154
154
in_channels .append (layer .in_channels )
155
155
out_channels .append (layer .out_channels )
156
156
return in_channels ,out_channels ,[]
157
-
158
157
elif model_name == 'resnet50' :
159
158
in_channels = []
160
159
out_channels = []
@@ -166,6 +165,7 @@ def net_info(model_name):
166
165
in_channels .append (layer .in_channels )
167
166
out_channels .append (layer .out_channels )
168
167
return in_channels ,out_channels ,[]
168
+
169
169
def create_edge_features (edge_types ,type_features ,device ):
170
170
if max (edge_types )> len (type_features ):
171
171
#random initial primitive operation like batch norm
Original file line number Diff line number Diff line change 27
27
- Rewrite the Graph Environment : empty.md
28
28
29
29
- About Us :
30
- - SwAPP : swapp.md
30
+ # - SwAPP: swapp.md
31
+ - SwAPP : empty.md
31
32
- API :
32
33
- gnnrl.lib : empty.md
33
34
- gnnrl.graph_env : empty.md
You can’t perform that action at this time.
0 commit comments