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'