Kevin Fink
commited on
Commit
·
85a2944
1
Parent(s):
648a87e
deve
Browse files
app.py
CHANGED
@@ -36,7 +36,8 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
|
|
36 |
preds, labels = eval_preds
|
37 |
if isinstance(preds, tuple):
|
38 |
preds = preds[0]
|
39 |
-
|
|
|
40 |
# Replace -100s used for padding as we can't decode them
|
41 |
preds = np.where(preds != -100, preds, tokenizer.pad_token_id)
|
42 |
labels = np.where(labels != -100, labels, tokenizer.pad_token_id)
|
|
|
36 |
preds, labels = eval_preds
|
37 |
if isinstance(preds, tuple):
|
38 |
preds = preds[0]
|
39 |
+
from pprint import pprint as pp
|
40 |
+
pp(preds)
|
41 |
# Replace -100s used for padding as we can't decode them
|
42 |
preds = np.where(preds != -100, preds, tokenizer.pad_token_id)
|
43 |
labels = np.where(labels != -100, labels, tokenizer.pad_token_id)
|