Fix checkpoint of Light-NAS (#18332)

Socket can't be pickled.

test=release/1.5

python/paddle/fluid/contrib/slim/nas/light_nas_strategy.py

@@ -127,6 +127,14 @@ def on_compression_begin(self, context):
         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)