Update the model inference logging message
Browse files- 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(
|
|
|
|
|
|
|
| 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(
|