diff --git a/python/paddle/fluid/contrib/slim/core/config.py b/python/paddle/fluid/contrib/slim/core/config.py index 9b08a0324a..cbe1c736fe 100644 --- a/python/paddle/fluid/contrib/slim/core/config.py +++ b/python/paddle/fluid/contrib/slim/core/config.py @@ -27,7 +27,7 @@ __all__ = ['ConfigFactory'] """This factory is used to create instances by loading and parsing configure file with yaml format. """ -PLUGINS = ['pruners', 'quantizers', 'quantizers', 'strategies', 'controllers'] +PLUGINS = ['pruners', 'quantizers', 'distillers', 'strategies', 'controllers'] class ConfigFactory(object): diff --git a/python/paddle/fluid/contrib/slim/core/strategy.py b/python/paddle/fluid/contrib/slim/core/strategy.py index f2cd2a2835..c0ddb758f3 100644 --- a/python/paddle/fluid/contrib/slim/core/strategy.py +++ b/python/paddle/fluid/contrib/slim/core/strategy.py @@ -29,6 +29,13 @@ class Strategy(object): self.start_epoch = start_epoch self.end_epoch = end_epoch + def __getstate__(self): + d = {} + for key in self.__dict__: + if key not in ["start_epoch", "end_epoch"]: + d[key] = self.__dict__[key] + return d + def on_compression_begin(self, context): pass