Repository: incubator-singa Updated Branches: refs/heads/master a94fa10fe -> 913417ad9
SINGA-327 Fix the bug of loading model saved in bin file Project: http://git-wip-us.apache.org/repos/asf/incubator-singa/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-singa/commit/913417ad Tree: http://git-wip-us.apache.org/repos/asf/incubator-singa/tree/913417ad Diff: http://git-wip-us.apache.org/repos/asf/incubator-singa/diff/913417ad Branch: refs/heads/master Commit: 913417ad9ae43871591a32325d5374baa46d6a75 Parents: a94fa10 Author: dbxinj <[email protected]> Authored: Tue Jul 4 13:54:41 2017 +0800 Committer: dbxinj <[email protected]> Committed: Tue Jul 4 13:54:41 2017 +0800 ---------------------------------------------------------------------- python/singa/net.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-singa/blob/913417ad/python/singa/net.py ---------------------------------------------------------------------- diff --git a/python/singa/net.py b/python/singa/net.py index 82933e1..3bd960a 100644 --- a/python/singa/net.py +++ b/python/singa/net.py @@ -414,10 +414,10 @@ class FeedForwardNet(object): otherwise, it would use protobuf for serialization, which uses less space. ''' + params['SINGA_VERSION'] = __version__ if use_pickle: params = {} # since SINGA>=1.1.1 (1101) - params['SINGA_VERSION'] = __version__ for (name, val) in zip(self.param_names(), self.param_values()): val.to_host() params[name] = tensor.to_numpy(val)
