AMR-KELEG commited on
Commit
84916fc
·
1 Parent(s): e7de010

Update the model inference logging message

Browse files
Files changed (1) hide show
  1. background_inference.py +3 -1
background_inference.py CHANGED
@@ -24,7 +24,9 @@ for i, sentence in enumerate(sentences):
24
  predictions.append(
25
  getattr(eval_utils, inference_function)(model, tokenizer, sentence)
26
  )
27
- print("Inference progress: ", round(100 * (i + 1) / len(sentences), 1), "%")
 
 
28
 
29
  # Store the predictions in a private dataset
30
  utils.upload_predictions(
 
24
  predictions.append(
25
  getattr(eval_utils, inference_function)(model, tokenizer, sentence)
26
  )
27
+ print(
28
+ f"Inference progress ({model_name}, {inference_function}): {round(100 * (i + 1) / len(sentences), 1)}%"
29
+ )
30
 
31
  # Store the predictions in a private dataset
32
  utils.upload_predictions(