dongjunwei.djw wenmeng.zwm 3 years ago
parent
commit
0703798fc2
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      maas_lib/trainers/nlp/sequence_classification_trainer.py

+ 1
- 1
maas_lib/trainers/nlp/sequence_classification_trainer.py View File

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


Loading…
Cancel
Save