diff options
author | ZJaume <jzaragoza@prompsit.com> | 2022-07-25 15:47:12 +0300 |
---|---|---|
committer | ZJaume <jzaragoza@prompsit.com> | 2022-07-25 15:47:12 +0300 |
commit | a5f8499287e38b1852a38d850c85f05ff7276d97 (patch) | |
tree | 64443448acc038d5f18603b3cdda9efb6c63da8e | |
parent | ce01ef0f75d3dcf2add2f8ef0e84053ea8fd8ac9 (diff) |
Redirect predict progbar to stder in debug mode
-rw-r--r-- | bicleaner_ai/models.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/bicleaner_ai/models.py b/bicleaner_ai/models.py index ef34308..c5d7ea6 100644 --- a/bicleaner_ai/models.py +++ b/bicleaner_ai/models.py @@ -219,7 +219,8 @@ class BaseModel(ModelInterface): generator = self.get_generator(batch_size, shuffle=False) generator.load((x1, x2, None)) - y_pred = self.model.predict(generator, verbose=verbose) + with redirect_stdout(sys.stderr): + y_pred = self.model.predict(generator, verbose=verbose) # Obtain logits if model returns HF output if isinstance(y_pred, TFSequenceClassifierOutput): y_pred = y_pred.logits |