SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix nan in optimizer_on_cpu

Browse Source
This commit is contained in:
thomwolf
2018-11-15 21:47:41 +01:00
parent 1de35b624b
commit cba85a67b9
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/run_squad.py
View File

@@ -687,6 +687,7 @@ def set_optimizer_params_grad(named_params_optimizer, named_params_model, test_n
if name_opti != name_model: if name_opti != name_model:
logger.error("name_opti != name_model: {} {}".format(name_opti, name_model)) logger.error("name_opti != name_model: {} {}".format(name_opti, name_model))
raise ValueError raise ValueError
if param_model.grad is not None:
if test_nan and torch.isnan(param_model.grad).sum() > 0: if test_nan and torch.isnan(param_model.grad).sum() > 0:
is_nan = True is_nan = True
if param_opti.grad is None: if param_opti.grad is None: