From 9624c366c5a3143eae0c6da778b716bfc0229088 Mon Sep 17 00:00:00 2001
From: liuzxtest02 <134442@163.com>
Date: Tue, 14 Nov 2023 15:16:40 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'gpu=5Fmnist=5Fexample/tra?=
 =?UTF-8?q?in=5Fgcu.py'?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 gpu_mnist_example/train_gcu.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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:")