GNN model

[sana-taj-79-26]

Training loop

for epoch in range(epochs): model.train() for data in train_loader: optimizer.zero_grad() output = model(data) loss = criterion(output, labels) loss.backward() optimizer.step() print(f"loss: {loss}" plz suggest some fix to it bcz it provide errors like AttributeError: 'Tensor' object has no attribute 'edge_index'