Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-05-18 05:05:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: UT error

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/8899458

    * fix: UT error
This commit is contained in:
wenmeng.zwm
2022-06-01 09:16:39 +08:00
committed by yingda.chen
parent 5995cc4607
commit 1d01a78c2b
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
configs/nlp/sequence_classification_trainer.yaml
View File

@@ -51,7 +51,7 @@ train:
num_steps: 100000
evaluation: # [being used]
model_path: .cache/easynlp/bert-base-sst2
model_path: .cache/easynlp/
max_sequence_length: 128
batch_size: 32
metrics:

4
tests/pipelines/test_text_classification.py
View File

@@ -42,7 +42,7 @@ class SequenceClassificationTest(unittest.TestCase):
with zipfile.ZipFile(cache_path_str, 'r') as zipf:
zipf.extractall(cache_path.parent)
path = r'.cache/easynlp/bert-base-sst2'
path = r'.cache/easynlp/'
model = SequenceClassificationModel(path)
preprocessor = SequenceClassificationPreprocessor(
path, first_sequence='sentence', second_sequence=None)
@@ -74,7 +74,7 @@ class SequenceClassificationTest(unittest.TestCase):
with zipfile.ZipFile(cache_path_str, 'r') as zipf:
zipf.extractall(cache_path.parent)
path = r'.cache/easynlp/bert-base-sst2'
path = r'.cache/easynlp/'
model = SequenceClassificationModel(path)
preprocessor = SequenceClassificationPreprocessor(
path, first_sequence='sentence', second_sequence=None)