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): ...@@ -41,7 +41,7 @@ class BaseRGCN(nn.Module):
def build_input_layer(self): def build_input_layer(self):
return None return None
def build_hidden_layer(self): def build_hidden_layer(self, idx):
raise NotImplementedError raise NotImplementedError
def build_output_layer(self): def build_output_layer(self):
......
...@@ -232,8 +232,8 @@ class Model(nn.Module): ...@@ -232,8 +232,8 @@ class Model(nn.Module):
i2h = self.build_input_layer() i2h = self.build_input_layer()
self.layers.append(i2h) self.layers.append(i2h)
# hidden to hidden # hidden to hidden
for idx in range(self.num_hidden_layers): for _ in range(self.num_hidden_layers):
h2h = self.build_hidden_layer(idx) h2h = self.build_hidden_layer()
self.layers.append(h2h) self.layers.append(h2h)
# hidden to output # hidden to output
h2o = self.build_output_layer() 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