yunjey / pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers
MIT License
29.54k stars 8k forks source link

GNN model #250

Open sana-taj-79-26 opened 7 months ago

sana-taj-79-26 commented 7 months ago

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'