Unverified Commit d5ae1ea0 authored by paoxiaode's avatar paoxiaode Committed by GitHub
Browse files

[BugFix] add input/output shape check in GatedGraphConv (#6119)


Co-authored-by: default avatarHongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
parent 66656d9d
...@@ -60,6 +60,7 @@ class GatedGraphConv(nn.Module): ...@@ -60,6 +60,7 @@ class GatedGraphConv(nn.Module):
def __init__(self, in_feats, out_feats, n_steps, n_etypes, bias=True): def __init__(self, in_feats, out_feats, n_steps, n_etypes, bias=True):
super(GatedGraphConv, self).__init__() super(GatedGraphConv, self).__init__()
assert in_feats <= out_feats, "out_feats must be not less than in_feats"
self._in_feats = in_feats self._in_feats = in_feats
self._out_feats = out_feats self._out_feats = out_feats
self._n_steps = n_steps self._n_steps = n_steps
......
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