from local

This commit is contained in:
dohe0342 2023-02-03 00:57:17 +09:00
parent 3858fccb33
commit 0e0714154a
2 changed files with 1 additions and 1 deletions

View File

@ -523,7 +523,7 @@ def train_one_epoch(
optimizer.zero_grad() optimizer.zero_grad()
loss.backward() loss.backward()
clip_grad_norm_(model.parameters(), 5.0, 2.0) #clip_grad_norm_(model.parameters(), 5.0, 2.0)
optimizer.step() optimizer.step()
if batch_idx % params.log_interval == 0: if batch_idx % params.log_interval == 0: