在燃烧的Pythorch之后进行跟踪和可视化
torchember的Python项目详细描述
火炬余烬
Tracking and visualize after the burning pytorch
该框架跟踪Pythorch模型:
- 在
nn.Module
级别 - 下至所有张量的度量/特征,包括
- 每个模块的输入/输出
- 重量/梯度张量
- 通过minimal额外编码
其他可爱的特征
- 可定制的指标,使用简单的装饰语法
- 以您喜欢的方式拆分跟踪日志,只需
mark(k=v,k1=v2...)
- 您可以轻松打开/关闭跟踪:
- 即使计算成本很小,torchember也不必为每次迭代计算度量
- 因此,你可以跟踪例如,只有最后一步,只有每200步。等等
安装
pip install torchember
快速教程
- 在
30秒tutorial
在 - 在
{a2满}
在
第一步,跟踪你的模型
把你的火把余烬追踪器放在你的模型上
fromtorchember.coreimporttorchEmberte=torchEmber(model)
上面可以跟踪每个模块的输入和输出,下面可以跟踪每个模块的状态
^{pr2}$像往常一样训练你的模特
步骤2,检查WebUI上的分析
从终端运行服务
$ torchember
默认端口为8080
或者分配一个端口
$ torchember --port=4200
请访问您的分析网站http://[host]:[port]
- 项目
标签: