Skip to content

GitLab

Commit dc5f7414 authored by rusty1s's avatar rusty1s
Browse files

fix

parent 9309d91b
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
examples/train_gcn2.py
View file @ dc5f7414
...@@ -60,7 +60,6 @@ def train(model, loader, optimizer): ...@@ -60,7 +60,6 @@ def train(model, loader, optimizer):
train_mask = batch.train_mask[:out.size(0)] train_mask = batch.train_mask[:out.size(0)]
loss = criterion(out[train_mask], batch.y[:out.size(0)][train_mask]) loss = criterion(out[train_mask], batch.y[:out.size(0)][train_mask])
loss.backward() loss.backward()
torch.nn.utils.clip_grad_norm_(model.parameters(), 1.0)
optimizer.step() optimizer.step()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment