|
@ -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) |
|
|
|
|
|
|
|
|