diff --git a/PaddleRec/gnn/train.py b/PaddleRec/gnn/train.py
index d82db868f6068110d81c99dd4cc0efbb471c5fa5..a470e62895f590e912b3fc647ac7a5c6a26fd361 100644
--- a/PaddleRec/gnn/train.py
+++ b/PaddleRec/gnn/train.py
@@ -22,6 +22,7 @@ import paddle.fluid as fluid
 import argparse
 import network
 import reader
+import sys
 
 logging.basicConfig(format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger("fluid")