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

---
 gpu_mnist_example/train_gpu.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/gpu_mnist_example/train_gpu.py b/gpu_mnist_example/train_gpu.py
index 31b026d..3ad5edf 100644
--- a/gpu_mnist_example/train_gpu.py
+++ b/gpu_mnist_example/train_gpu.py
@@ -91,6 +91,10 @@ if __name__ == '__main__':
     print(os.listdir(pretrain_model_path))
     os.listdir(pretrain_model_path)
 
+    print("output_path:")
+    print(os.listdir(output_path))
+    os.listdir(output_path)
+
     #log output
     print('cuda is available:{}'.format(torch.cuda.is_available()))  
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")