YooChoose challenge with GCN in Pytorch Geometric