DGLError: Cannot update column of scheme Scheme(shape=(16,), dtype=torch.float32) using feature of scheme Scheme(shape=(2,), dtype=torch.float32)

[TungTNguyen]

I'm reproducing this paper and come across this bug, while running both main.py and parameter_reload.py.
This could have been resulted from this line of code:
self.gcn1 = GCN(2, 16, t.tanh)
which initialize a Graph Convolutional Networks with input_features=2, out_features=16 and activation=tanh. Any debugging would be acknowledged. @arbaazkhan2