神经网络可视化新工具:TorchExplorer

发布时间 2023-12-22 10:11:13作者: deephub

TorchExplorer是一个交互式探索神经网络的可视化工具,他的主要功能如下:

TorchExplorer是一款创新的人工智能工具,专为使用非常规神经网络架构的研究人员设计。可以在本地或者wandb中生成交互式Vega自定义图表,提供网络结构的模块级可视化。在左边的面板可以模块级方式展现神经网络架构,帮助研究人员导航网络结构。在右边的图中节点表示输入/输出占位符或在转发过程中调用的特定子模块,可以深入检查模块,直方图可视化数据分布。

节点之间的边缘表示数据处理流向,并且提供对输入/输出张量,梯度规范和参数梯度的信息。

https://avoid.overfit.cn/post/e7be6a62915445e8ab8af0b40f13019a