mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-13 08:28:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Anurag Lamsal
2024-09-24 09:05:44 +05:45
committed by GitHub
parent 4a2fa0b627
commit 568757e087
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/mlperf/model_eval.py
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