nnzhan / Graph-WaveNet

graph wavenet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Bug in class gwnet(nn.Module)

lrk612 opened this issue · comments

len(supports) in self.gconv.append(gcn(dilation_channels,residual_channels,dropout,support_len=len(supports))) should be replaced as self.supports_len.
Otherwise when no supports are passed in, since self.supports is an empty list, it will cause the dimension c_in of the mlp in gcn to not match the data. I think this is why the author specifically maintained the variable self.supports_len earlier and +1