diff --git a/python/paddle/fluid/contrib/slim/core/compressor.py b/python/paddle/fluid/contrib/slim/core/compressor.py
index 21737c76251704d739d588db755734705ca7601c..41f282bf997e0bc925920044c88a8e1e57b54c11 100644
--- a/python/paddle/fluid/contrib/slim/core/compressor.py
+++ b/python/paddle/fluid/contrib/slim/core/compressor.py
@@ -55,7 +55,9 @@ def cached_reader(reader, sampled_rate, cache_path, cached_id):
     def s_reader():
         if os.path.isdir(cache_path):
             for file_name in open(os.path.join(cache_path, "list")):
-                yield np.load(os.path.join(cache_path, file_name.strip()))
+                yield np.load(
+                    os.path.join(cache_path, file_name.strip()),
+                    allow_pickle=True)
         else:
             os.makedirs(cache_path)
             list_file = open(os.path.join(cache_path, "list"), 'w')