Kevin Fink commited on
Commit
85a2944
·
1 Parent(s): 648a87e
Files changed (1) hide show
  1. app.py +2 -1
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)