diff --git a/gpu_mnist_example/train_gcu.py b/gpu_mnist_example/train_gcu.py
index eb52a2a..c9e97f6 100644
--- a/gpu_mnist_example/train_gcu.py
+++ b/gpu_mnist_example/train_gcu.py
@@ -74,7 +74,7 @@ if __name__ == '__main__':
     pretrain_model_path = openi_context.pretrain_model_path
     output_path = openi_context.output_path
 
-    dataset_path_A = dataset_path + "/MnistDataset_torch"
+    dataset_path_A = dataset_path + "/MnistDataset"
     pretrain_model_path_A = pretrain_model_path + "/MNIST_PytorchExample_GPU_test34_model_7f9j"
 
     print("dataset_path:")