diff --git a/research/nlp/gpt2/convert_tf_ckpt/save_weight_ms.py b/research/nlp/gpt2/convert_tf_ckpt/save_weight_ms.py
index dcb4da90e2068879d203ced062352cc245b446bb..40b5f10f17851b9928e9a6408d1980ea8cc0794c 100644
--- a/research/nlp/gpt2/convert_tf_ckpt/save_weight_ms.py
+++ b/research/nlp/gpt2/convert_tf_ckpt/save_weight_ms.py
@@ -23,7 +23,7 @@ import numpy as np
 from mindspore import Tensor
 from mindspore.train.serialization import save_checkpoint
 
-from .trans_dict import trans_dict_tf
+from trans_dict import trans_dict_tf
 
 
 def trans_model_parameter(ckpt_name):