From 0703798fc2edc579b9b2a092acd0c51cb2f248c0 Mon Sep 17 00:00:00 2001 From: "dongjunwei.djw" Date: Tue, 31 May 2022 11:44:11 +0800 Subject: [PATCH] fix the model evaluation error Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/8872484 --- maas_lib/trainers/nlp/sequence_classification_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/maas_lib/trainers/nlp/sequence_classification_trainer.py b/maas_lib/trainers/nlp/sequence_classification_trainer.py index e88eb95e..f2264c0d 100644 --- a/maas_lib/trainers/nlp/sequence_classification_trainer.py +++ b/maas_lib/trainers/nlp/sequence_classification_trainer.py @@ -128,7 +128,7 @@ class SequenceClassificationTrainer(BaseTrainer): collate_fn=pre_dataset.batch_fn) # generate a model - model = SequenceClassification(checkpoint_path) + model = SequenceClassification.from_pretrained(checkpoint_path) # copy from easynlp (start) model.eval()