fix

dc5f7414 · rusty1s · 9309d91b · dc5f7414
Commit dc5f7414 authored Jun 07, 2021 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

examples/train_gcn2.py examples/train_gcn2.py +0 -1

No files found.
--- a/examples/train_gcn2.py
+++ b/examples/train_gcn2.py
@@ -60,7 +60,6 @@ def train(model, loader, optimizer):
        train_mask = batch.train_mask[:out.size(0)]
        loss = criterion(out[train_mask], batch.y[:out.size(0)][train_mask])
        loss.backward()
-        torch.nn.utils.clip_grad_norm_(model.parameters(), 1.0)
        optimizer.step()