TopoGNN MUTAG error

[soham0209]

Hi, When running TopoGNN with poetry run python topognn/train_model.py --model TopoGNN --dataset MUTAG --max_epochs 10, the run fails with: /pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:108: cunn_ClassNLLCriterion_updateOutput_kernel: block: [0,0,0], thread: [0,0,0] Assertion t >= 0 && t < n_classes failed. I am almost sure that this is not an environment issue since other datasets such as DD, IMDB-MULTI runs successfully. My guess is that the labels for MUTAG is incorrect.