frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-06 00:54:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix serialization of nn.Parameter with dill (#10296)

Browse Source 
Summary:
Should resolve #9981.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/10296

Differential Revision: D9196353

Pulled By: soumith

fbshipit-source-id: 109b6da42b7240cdbc7a0586745c735bce5e1279
This commit is contained in:
Marcin Elantkowski
2018-09-01 23:53:32 -07:00
committed by Facebook Github Bot
parent 1350f76b62
commit 4d28b65fb8
2 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
torch/_utils.py
View File

@ -101,6 +101,13 @@ def _rebuild_tensor_v2(storage, storage_offset, size, stride, requires_grad, bac
return tensor
def _rebuild_parameter(data, requires_grad, backward_hooks):
param = torch.nn.Parameter(data, requires_grad)
param._backward_hooks = backward_hooks
return param
def _import_dotted_name(name):
components = name.split('.')
obj = __import__(components[0])