Update classifier.py

This commit is contained in:
smthemex
2024-08-11 16:23:23 +08:00
committed by GitHub
parent b50a276b7a
commit a562b5a6b6
+1 -1
View File
@@ -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,