My index is out of bounds for some graph in gcn model

I am trying to do a graph convolution network with forward-forward propagation. When I am running the model, I get an error saying that my index is out of bounds, but this happens after the model works fine for some examples. My dataset contains 253 graphs and the batch size is 64. I don’t understand what would be the problem for one graph but not for the others. Thank you! import torch import torch.nn.functional as F import subprocess import torch.nn as nn from torch.nn import MSELoss from tor...