From 6ecf08b17395944febedd9867c48000a9d7a9f9f Mon Sep 17 00:00:00 2001 From: guosheng Date: Tue, 12 Dec 2017 19:39:07 +0800 Subject: [PATCH 1/2] Enhance init_from_tar to support indicating parameters excluded from the initialized model --- python/paddle/v2/parameters.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/paddle/v2/parameters.py b/python/paddle/v2/parameters.py index bd97dc1199..0dd7d38d8e 100644 --- a/python/paddle/v2/parameters.py +++ b/python/paddle/v2/parameters.py @@ -383,19 +383,21 @@ class Parameters(object): params.deserialize(param_name, f) return params - def init_from_tar(self, f): + def init_from_tar(self, f, exclude_params=[]): """ Different from `from_tar`, this interface can be used to init partial network parameters from another saved model. :param f: the initialized model file. :type f: tar file + :param exclude_params: the names of parameters that shouldn't be initialized from the model file. + :type exclude_params: list of strings :return: Nothing. """ tar_param = Parameters.from_tar(f) for pname in tar_param.names(): - if pname in self.names(): + if pname in self.names() and pname not in exclude_params: self.set(pname, tar_param.get(pname)) From 2cc34532437cb73c00cfe77facb63cb3578ae651 Mon Sep 17 00:00:00 2001 From: guosheng Date: Wed, 13 Dec 2017 10:58:50 +0800 Subject: [PATCH 2/2] Refine the doc of parameters.init_from_tar --- python/paddle/v2/parameters.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/v2/parameters.py b/python/paddle/v2/parameters.py index 0dd7d38d8e..7b7d1a1d16 100644 --- a/python/paddle/v2/parameters.py +++ b/python/paddle/v2/parameters.py @@ -390,7 +390,8 @@ class Parameters(object): :param f: the initialized model file. :type f: tar file - :param exclude_params: the names of parameters that shouldn't be initialized from the model file. + :param exclude_params: the names of parameters that should + not be initialized from the model file. :type exclude_params: list of strings :return: Nothing. """