文件名称:PyTorch的tensorboard插件-python开发
文件大小:4.66MB
文件格式:ZIP
更新时间:2024-07-08 11:45:52
机器学习
PyTorch的tensorboard插件,只用很简单的函数调用就能写出Tensorboard事件,支持模型图可视化。除了PyTorch外,Chainer、Mxnet和Numpy也适用。 tensorboardX 使用简单的函数调用编写 TensorBoard 事件。 支持标量、图像、图形、直方图、音频、文本、图形、onnx_graph、embedding、pr_curve、mesh、超参数和视频摘要。 demo_graph.py 的要求是 tensorboardX>=1.9 和 pytorch>=1.3 常见问题安装 在 anaconda2 / anaconda3 上测试,使用 PyTorch 1.3.1 / torchvision 0.4.1 / tensorboard 2.0.0 pip install tensorboardX 或从源代码构建:pip install 'git +https://github.com/lanpa/tensorboardX' 您可以选择安装 crc32c 以加快保存大量数据的速度。 示例 克隆 https://github.co