From a562b5a6b615187f75f0aa98ac77e1f6a23654f8 Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Sun, 11 Aug 2024 16:23:23 +0800
Subject: [PATCH] Update classifier.py

---
 custom_net/classifier.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/custom_net/classifier.py b/custom_net/classifier.py
index 6bf9b09..177a1aa 100644
--- a/custom_net/classifier.py
+++ b/custom_net/classifier.py
@@ -11,7 +11,7 @@ from glob import glob
 from natsort import natsorted
 
 from .openaimodel import EncoderUNetModel, UNetModel
-from customnet_util import log_txt_as_img, default, ismap, instantiate_from_config
+from .customnet_util import log_txt_as_img, default, ismap, instantiate_from_config
 
 __models__ = {
     'class_label': EncoderUNetModel,