更新 'gpu_mnist_example/train_gpu.py'

gpu_mnist_example/train_gpu.py

@@ -84,6 +84,11 @@ if __name__ == '__main__':
     pretrain_model_path = openi_context.pretrain_model_path
     output_path = openi_context.output_path
 
+    print("test dataset:")
+    os.listdir(dataset_path)
+    print("test model:")
+    os.listdir(pretrain_model_path)
+
     #log output
     print('cuda is available:{}'.format(torch.cuda.is_available()))  
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")