更新 'gpu_mnist_example/train_gpu.py'

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