When I run this Notebook https://github.com/khuangaf/PyTorch-Geometric-YooChoose/blob/master/YooChooseBuy.ipynb to learn Graph Neural Networks, I get the error ValueError: too many values to unpack (expected 5). I am new to PyTorch and graph neural networks. It seems this code worked before. I navigated the code and found the output is not right. It’s not a tuple or list with five elements. Can anyone tell me how to fix this?
torch.Size([2093, 256])
torch.Size([2093, 128])
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
<ipython-input-49-fb44712a6f41> in <module>
1 for epoch in range(1, 2):
----> 2 loss = train()
3 train_acc = evaluate(train_loader)
4 val_acc = evaluate(val_loader)
5 test_acc = evaluate(test_loader)
<ipython-input-47-d6d407b09375> in train()
6 data = data.to(device)
7 optimizer.zero_grad()
----> 8 output = model(data)
9
10 label = data.y.to(device)
C:\ProgramData\Miniconda3\envs\ptg\lib\site-packages\torch\nn\modules\module.py in _call_impl(self, *input, **kwargs)
720 result = self._slow_forward(*input, **kwargs)
721 else:
--> 722 result = self.forward(*input, **kwargs)
723 for hook in itertools.chain(
724 _global_forward_hooks.values(),
<ipython-input-45-312d18271bec> in forward(self, data)
38 x = F.relu(self.conv1(x, edge_index))
39 print(x.shape)
---> 40 x, edge_index, _, batch, _ = self.pool1(x, edge_index, None, batch)
41 # x = self.pool1(x, edge_index, None, batch)
42 x1 = torch.cat([gmp(x, batch), gap(x, batch)], dim=1)
ValueError: too many values to unpack (expected 5)