Unverified Commit ae1806f6 authored by Zihao Ye's avatar Zihao Ye Committed by GitHub
Browse files

[Bugfix] RGCN tutorial (#388)

* fix rgcn tutorial

* small fix

* upd
parent 59ec7a9c
......@@ -41,7 +41,7 @@ class BaseRGCN(nn.Module):
def build_input_layer(self):
return None
def build_hidden_layer(self):
def build_hidden_layer(self, idx):
raise NotImplementedError
def build_output_layer(self):
......
......@@ -232,8 +232,8 @@ class Model(nn.Module):
i2h = self.build_input_layer()
self.layers.append(i2h)
# hidden to hidden
for idx in range(self.num_hidden_layers):
h2h = self.build_hidden_layer(idx)
for _ in range(self.num_hidden_layers):
h2h = self.build_hidden_layer()
self.layers.append(h2h)
# hidden to output
h2o = self.build_output_layer()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment