algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-05 16:36:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

clip_grad_norm is now deprecated

Browse Source
This commit is contained in:
e
2018-11-09 16:21:42 +08:00
parent c48008bf9d
commit a16173ffc4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tutorials/02-intermediate/language_model/main.py
View File

@ -3,7 +3,7 @@
import torch
import torch.nn as nn
import numpy as np
from torch.nn.utils import clip_grad_norm
from torch.nn.utils import clip_grad_norm_
from data_utils import Dictionary, Corpus
@ -78,7 +78,7 @@ for epoch in range(num_epochs):
# Backward and optimize
model.zero_grad()
loss.backward()
clip_grad_norm(model.parameters(), 0.5)
clip_grad_norm_(model.parameters(), 0.5)
optimizer.step()
step = (i+1) // seq_length