|
|
|
@@ -13,7 +13,7 @@ from torch import nn |
|
|
|
from torch.optim import SGD |
|
|
|
from torch.optim.lr_scheduler import StepLR |
|
|
|
|
|
|
|
from modelscope.metainfo import Trainers |
|
|
|
from modelscope.metainfo import Metrics, Trainers |
|
|
|
from modelscope.metrics.builder import MetricKeys |
|
|
|
from modelscope.msdatasets import MsDataset |
|
|
|
from modelscope.trainers import build_trainer |
|
|
|
@@ -102,7 +102,7 @@ class TrainerTest(unittest.TestCase): |
|
|
|
'workers_per_gpu': 1, |
|
|
|
'shuffle': False |
|
|
|
}, |
|
|
|
'metrics': ['seq-cls-metric'] |
|
|
|
'metrics': [Metrics.seq_cls_metric] |
|
|
|
} |
|
|
|
} |
|
|
|
config_path = os.path.join(self.tmp_dir, ModelFile.CONFIGURATION) |
|
|
|
@@ -156,7 +156,7 @@ class TrainerTest(unittest.TestCase): |
|
|
|
'workers_per_gpu': 1, |
|
|
|
'shuffle': False |
|
|
|
}, |
|
|
|
'metrics': ['seq-cls-metric'] |
|
|
|
'metrics': [Metrics.seq_cls_metric] |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
@@ -206,7 +206,7 @@ class TrainerTest(unittest.TestCase): |
|
|
|
'workers_per_gpu': 1, |
|
|
|
'shuffle': False |
|
|
|
}, |
|
|
|
'metrics': ['seq-cls-metric'] |
|
|
|
'metrics': [Metrics.seq_cls_metric] |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|