pytorch/ignite
Warning using ConfusionMatrix and setup_tb_logging
Geschlossen
#829 geöffnet am 04.03.2020
enhancementhelp wantedmodule: metrics
Repository-Metriken
- Stars
- (4.313 Sterne)
- PR-Merge-Metriken
- (Durchschn. Merge 18T 23h) (19 gemergte PRs in 30 T)
Beschreibung
❓ 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 ?