diff --git a/python/paddle/fluid/contrib/slim/nas/light_nas_strategy.py b/python/paddle/fluid/contrib/slim/nas/light_nas_strategy.py
index f5e9ca5ec42390617b69514d84897d81e9dc79f6..7f44510bd582cd503b230645fa02100d1a003819 100644
--- a/python/paddle/fluid/contrib/slim/nas/light_nas_strategy.py
+++ b/python/paddle/fluid/contrib/slim/nas/light_nas_strategy.py
@@ -127,6 +127,14 @@ class LightNASStrategy(Strategy):
         self._search_agent = SearchAgent(
             self._server_ip, self._server_port, key=self._key)
 
+    def __getstate__(self):
+        """Socket can't be pickled."""
+        d = {}
+        for key in self.__dict__:
+            if key not in ["_search_agent", "_server"]:
+                d[key] = self.__dict__[key]
+        return d
+
     def _constrain_func(self, tokens, context=None):
         """Check whether the tokens meet constraint."""
         _, _, test_prog, _, _, _, _ = context.search_space.create_net(tokens)