Nixtla / neuralforecast

Scalable and user friendly neural :brain: forecasting algorithms.
https://nixtlaverse.nixtla.io/neuralforecast
Apache License 2.0
3.13k stars 362 forks source link

nf.predict spam printing in prediction #1026

Closed dara1400 closed 5 months ago

dara1400 commented 5 months ago

What happened + What you expected to happen

I want to use nf.predict in a loop. It keeps printing this:

GPU available: True (cuda), used: True TPU available: False, using: 0 TPU cores IPU available: False, using: 0 IPUs HPU available: False, using: 0 HPUs LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0]

Is there way to stop it from printing?

Versions / Dependencies

Windows 11, Python 3.10, neuralforecast v1.7.2

Reproduction script

for i in range(100):
      nf.predict(x[i])

Issue Severity

High: It blocks me from completing my task.

jmoralez commented 5 months ago
import logging
logging.getLogger('pytorch_lightning').setLevel(logging.ERROR)