I am working on loading my own dataset using PyG

I am getting an error saying

index is out of bounds for dimension 0 with size 10

how to solve this? code:

for epoch in range(1,101):
    loss = train()
    print(f'Epoch: {epoch:03d}, Loss: {loss:.4f}')