Как заставить замолчать информацию о тренировках в GluonTS / Apache m xnet?

Question

Когда я тренирую свой DeepAREstimator в GluonTS

estimator = DeepAREstimator(freq="D", 
                              prediction_length=56, 
                              trainer=Trainer(epochs=20))

predictor = estimator.train(training_data=training_data)

, я получаю кучу отладочных сообщений:

INFO:root:Epoch[12] Learning rate is 0.001
100%|██████████| 50/50 [00:09<00:00,  5.31it/s, avg_epoch_loss=2.29]
INFO:root:Epoch[12] Elapsed time 9.422 seconds
INFO:root:Epoch[12] Evaluation metric 'epoch_loss'=2.293711
INFO:root:Epoch[13] Learning rate is 0.001

Как я могу заставить их замолчать? Это может быть связано с лежащим в основе m xnet, поэтому я тоже отмечу это. Может кто-нибудь добавить тег глюонцев?

Answer 1

import logging

logging.getLogger("mxnet").addFilter(lamda record: False)