change import level

2 years ago · 8be5baa9b2
1 changed files with 1 additions and 2 deletions
--- 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)