pytorch中tensorboard网络可视化输出错误：

最新推荐文章于 2023-07-26 11:38:37 发布

原创

最新推荐文章于 2023-07-26 11:38:37 发布 · 2.1k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #python #人工智能

本文介绍如何在PyTorch模型中遇到dict输出时，通过使用NamedTuple来解决可视化问题。作者提供了一个ModelWrapper类示例，指导读者如何将模型的dict输出转换为tuple或NamedTuple，以确保正确显示并避免Tracer错误。

对于模型的输出是list或者dict，需要使用可视化展示model时会产生错误，仅支持tuple或者namedtuple输出方式：

RuntimeError: Encountering a dict at the output of the tracer might cause the trace to be incorrect, this is only valid if the container structure does not change based on the module's inputs. Consider using a constant container instead (e.g. for `list`, use a `tuple` instead. for `dict`, use a `NamedTuple` instead). If you absolutely need this and know the side effects, pass strict=False to trace() to allow this behavior.

可以通过如下方法将模型的输出进行更改：

from collections import namedtuple
from typing import Any

# pylint: disable = abstract-method
class ModelWrapper(torch.nn.Module):
    """
    Wrapper class for model with dict/list rvalues.
    """

    def __init__(self, model: torch.nn.Module) -> None:
        """
        Init call.
        """
        super().__init__()
        self.model = model