From 73c952202f68215820b1f40c319beeb0ad0b976c Mon Sep 17 00:00:00 2001
From: wrz1-zzzzz <914921336@qq.com>
Date: Fri, 15 Nov 2024 10:22:41 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 DPL/dgd_class/predict.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/DPL/dgd_class/predict.py b/DPL/dgd_class/predict.py
index 2d72872..679f0fe 100644
--- a/DPL/dgd_class/predict.py
+++ b/DPL/dgd_class/predict.py
@@ -8,7 +8,7 @@ from torchvision import datasets, models, transforms
 import os
 import matplotlib.pyplot as plt
 
-from test import device, class_names, imshow
+from train import device, class_names, imshow
 
 
 # 加载已训练的 ONNX 模型