pytorch/ignite

Warning using ConfusionMatrix and setup_tb_logging

已关闭

#829 创建于 2020年3月4日

在 GitHub 查看贡献者指南
 (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 ?

贡献者指南

研究方向
Examine the setup tb logging function in ignite/contrib/engines/common.py (lines 248-254). The OutputHandler is instantiated with metric names="all", causing a warning when ConfusionMatrix returns tensors. Consider adding a parameter to setup tb logging to allow users to specify metric names, thereby avoiding the warning. Ensure backward compatibility and check if similar changes are needed in setup any logging.
技术栈
python
领域
machine learning
议题类型
功能
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
3
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
1-3 小时
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
较久未更新
清晰度议题是否清楚说明期望改动、验收标准和下一步。
清晰
前置要求
PythonPyTorchIgnite basics
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
70