From 2871dc103bd667824d750c603ffba09f03aae607 Mon Sep 17 00:00:00 2001
From: liuzxtest02 <134442@163.com>
Date: Mon, 6 Nov 2023 14:42:51 +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, 2 insertions(+), 2 deletions(-)

diff --git a/gpu_mnist_example/train_gpu.py b/gpu_mnist_example/train_gpu.py
index 3ad5edf..2b893c4 100644
--- a/gpu_mnist_example/train_gpu.py
+++ b/gpu_mnist_example/train_gpu.py
@@ -11,7 +11,8 @@ If there are Chinese comments in the code，please add at the beginning：
   └── train  
 
 '''
-
+import os
+print(os.listdir('/code'))
 
 from model import Model
 import numpy as np
@@ -22,7 +23,6 @@ from torch.optim import SGD
 from torch.utils.data import DataLoader
 from torchvision.transforms import ToTensor
 import argparse
-import os
 #导入openi包
 from openi.context import prepare, upload_openi