From 8be5baa9b2963db26c298a2509ea27206716dc65 Mon Sep 17 00:00:00 2001 From: ChengZi Date: Thu, 29 Dec 2022 19:34:17 +0800 Subject: [PATCH] change import level --- isc.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/isc.py b/isc.py index 4fbaf83..171cdd2 100644 --- a/isc.py +++ b/isc.py @@ -32,8 +32,6 @@ import timm import warnings -from .train_isc import train_isc - warnings.filterwarnings('ignore') log = logging.getLogger('isc_op') @@ -179,6 +177,7 @@ class Isc(NNOperator): train_dataset=None, eval_dataset=None, resume_checkpoint_path=None, **kwargs): + from .train_isc import train_isc training_args = kwargs.pop('training_args', None) train_isc(self._model, training_args)