Update test_metric.py

2024-03-29 00:46:15 +08:00 · 2024-03-29 00:46:15 +08:00 · 89f188d935
parent dd581460f2
commit 89f188d935
1 changed files with 1 additions and 1 deletions
--- a/BERT/test_metric.py
+++ b/BERT/test_metric.py
@ -42,7 +42,7 @@ if __name__ == '__main__':
    data_loader = DataLoader(dataset, batch_size=batch_size, shuffle=False, num_workers=1)

    # 加载模型和检查点
-    model = BertForSequenceClassification.from_pretrained('./BERT_training_output/checkpoint-35544')
+    model = BertForSequenceClassification.from_pretrained('./BERT_training_output/checkpoint-35544'， num_labels = 6)

    # 运行推断
    model.eval()