#50 · 主分类: MLOps与评测

tensorboardX

machine-learning numpy pytorch tensorboard visualization

tensorboard 用于 pytorch(以及 chainer、mxnet、numpy 等)

项目最后更新:07/14/26

GitHub Stars

8.0K

Forks数量

852

贡献者数量

86

许可证

MIT

收录理由

用PyTorch做训练时,不想只在终端里盯着loss数字跳动,可以用这个库把TensorBoard的仪表盘搬进自己的训练流程,完全不必安装TensorFlow。创建一个SummaryWriter,调用几个方法,就能把标量、图像、音频、嵌入向量、直方图以及模型结构写进标准的事件文件,直接用常规的tensorboard服务器读取。同一套API也兼容NumPy、Chainer和MXNet,不同训练框架共用一条日志通路。由于写的是标准事件格式,已有的仪表盘不用改动,当需要跨实验做更丰富的对比时,这些数据也能转发到云端追踪服务。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目