From 952d7e3a3c0657f6b4ceb1a8bea47fe5bb49d8d6 Mon Sep 17 00:00:00 2001
From: Brainway <brainway@protonmail.com>
Date: Tue, 18 Oct 2022 03:59:15 +0000
Subject: [PATCH] update testsingle.py.

---
 testsingle.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/testsingle.py b/testsingle.py
index e3ecfd9..a6f574f 100755
--- a/testsingle.py
+++ b/testsingle.py
@@ -36,7 +36,7 @@ model = None
 #model = VisionTransformer(config, args.img_size, zero_head=True, num_classes=num_classes, smoothing_value=args.smoothing_value)
 
 if args.pretrained_model is not None:
-    model = torch.load(args.pretrained_model) #自己预训练模型
+    model = torch.load(args.pretrained_model,map_location=torch.device('cpu')) #自己预训练模型
 model.to(args.device)
 model.eval()