diff --git a/tools/export_model.py b/tools/export_model.py
index 29a540d68b34ae8641c33163b9703142ccc42d00..969c730d4d7c9a132133a3c08f5be870d52317de 100644
--- a/tools/export_model.py
+++ b/tools/export_model.py
@@ -13,6 +13,11 @@
 # limitations under the License.
 
 import argparse
+import os
+import sys
+__dir__ = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(__dir__)
+sys.path.append(os.path.abspath(os.path.join(__dir__, '..')))
 
 from ppcls.modeling import architectures
 from ppcls.utils.save_load import load_dygraph_pretrain