Hide device info prints in Python about comet

[ricardorei]

You are right. It's just solved for the CLI interface. Ill reopen the issues as the current behaviour is still not the expected behaviour.

Meanwhile I'll leave here a snippet on how to run it in a quiet mode and silencing the underlying libraries.

import logging
loggers = [logging.getLogger(name) for name in logging.root.manager.loggerDict]
for logger in loggers:
    logger.setLevel(logging.WARNING)

Finally, you can run COMET with progress_bar=False :

model_output = model.predict(data, batch_size=8, gpus=1, progress_bar=False)

[ricardorei]

Both pytorch lightning and transformers are very verbose.

I am adding a --quiet flag that sets all loggers to ERROR level. It will be included in the next release

[AlexUmnov]

@ricardorei thanks for solving this issue a while back.
Is it possible to add it to the documentation, how to actually use this flag? I'm using version 2.0.0 from python and it's not obvious for me how to hide the device printouts.

[ricardorei]

Its in the README:

comet-score -s src.de -t hyp1.en -r ref.en --quiet --only_system

[AlexUmnov]

@ricardorei but the issue mentions calling comet from python. So am I. Not the cli

[AlexUmnov]

I'm also getting a pytorch_lightning warning once I implement the code you suggested. Everything else is gone.

PossibleUserWarning: `max_epochs` was not set. Setting it to 1000 epochs. To train without an epoch limit, set `max_epochs=-1`.
  rank_zero_warn(