diff --git a/gpu_mnist_example/train_gpu.py b/gpu_mnist_example/train_gpu.py
index 6cbad58..42c1c00 100644
--- a/gpu_mnist_example/train_gpu.py
+++ b/gpu_mnist_example/train_gpu.py
@@ -84,9 +84,11 @@ if __name__ == '__main__':
     pretrain_model_path = openi_context.pretrain_model_path
     output_path = openi_context.output_path
 
-    print("test dataset:")
+    print("dataset_path:")
+    print(os.listdir(dataset_path))
     os.listdir(dataset_path)
-    print("test model:")
+    print("pretrain_model_path:")
+    print(os.listdir(pretrain_model_path))
     os.listdir(pretrain_model_path)
 
     #log output