From 07b925a84c32bdf84326abafd4487a19031be3ab Mon Sep 17 00:00:00 2001 From: liuzxtest02 <134442@163.com> Date: Fri, 3 Nov 2023 17:27:53 +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 | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/gpu_mnist_example/train_gpu.py b/gpu_mnist_example/train_gpu.py index 7bdebcc..6cbad58 100644 --- a/gpu_mnist_example/train_gpu.py +++ b/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")