fix model_eval.py in the mlperf folder searching for bert vocab in the wrong directory (#6649)

This commit is contained in:
Anurag Lamsal 2024-09-24 09:05:44 +05:45 committed by GitHub
commit 568757e087
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -181,7 +181,7 @@ def eval_bert():
from examples.mlperf.metrics import f1_score
from transformers import BertTokenizer
tokenizer = BertTokenizer(str(Path(__file__).parents[2] / "weights/bert_vocab.txt"))
tokenizer = BertTokenizer(str(Path(__file__).parents[2] / "extra/weights/bert_vocab.txt"))
c = 0
f1 = 0.0