pytorch/ignite

Warning using ConfusionMatrix and setup_tb_logging

已關閉

#829 建立於 2020年3月4日

 (2 則留言) (0 個反應) (0 位負責人)Python (602 個分叉)batch import
enhancementhelp wantedmodule: metrics

倉庫指標

星標
 (4,313 顆星)
PR 合併指標
 (平均合併 18天 23小時) (30 天內合併 19 個 PR)

描述

❓ Questions/Help/Support

logger configured by setup_tb_logging warns when a ConfusionMatrix metric is attached to the evaluator

tensorboard_logger.py:129: UserWarning: TensorboardLogger output_handler can not log metrics value type <class 'torch.Tensor'>
  "TensorboardLogger output_handler can not log " "metrics value type {}".format(type(value))

The reason is setup_tb_logging (via setup_any_logging) defines OutputHandler using metric_names="all" https://github.com/pytorch/ignite/blob/a22a0f5e909ac70d2a1f76a60b6e84b2134f196c/ignite/contrib/engines/common.py#L248-L254

What do you think to expose metric_names in the API to be able to fix the warning ?

貢獻者指南