update examples from master

This commit is contained in:
thomwolf committed 2018-11-17 12:21:35 +01:00
commit 47a7d4ec14
3 files changed
+11 -6

No files matched your search

+2
View File
@@ -693,6 +693,8 @@ def set_optimizer_params_grad(named_params_optimizer, named_params_model, test_n
if param_opti.grad is None:
param_opti.grad = torch.nn.Parameter(param_opti.data.new().resize_(*param_opti.data.size()))
param_opti.grad.data.copy_(param_model.grad.data)
else:
param_opti.grad = None
return is_nan
def main():