logo
Browse Source

change import level

main
ChengZi 2 years ago
parent
commit
8be5baa9b2
  1. 3
      isc.py

3
isc.py

@ -32,8 +32,6 @@ import timm
import warnings import warnings
from .train_isc import train_isc
warnings.filterwarnings('ignore') warnings.filterwarnings('ignore')
log = logging.getLogger('isc_op') log = logging.getLogger('isc_op')
@ -179,6 +177,7 @@ class Isc(NNOperator):
train_dataset=None, train_dataset=None,
eval_dataset=None, eval_dataset=None,
resume_checkpoint_path=None, **kwargs): resume_checkpoint_path=None, **kwargs):
from .train_isc import train_isc
training_args = kwargs.pop('training_args', None) training_args = kwargs.pop('training_args', None)
train_isc(self._model, training_args) train_isc(self._model, training_args)

Loading…
Cancel
Save