Update classifier.py
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user