TopK pooling layer giving return error

AsmaAdil · April 5, 2020, 9:34am

I am following this tutorial

khuangaf/PyTorch-Geometric-YooChoose/blob/master/YooChooseBuy.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import pandas as pd\n",
    "import pickle\n",
    "import csv\n",
    "import os\n",
    "import torch\n",
    "from torch_geometric.data import Data\n",
    "from tqdm import tqdm\n",
    "np.random.seed(42)"
   ]
  },
  {

This file has been truncated. show original

line no 157
x, edge_index, _, batch, _ = self.pool1(x, edge_index, None, batch)
giving error. I thought may be it returns four tensors as output instead of five, even that also giving error

ptrblck · April 5, 2020, 10:14pm

I guess the note book might not have been updated to the latest PyTorch Geometric version, as TopKPooling returns not 6 values as seen in this line of code.