在燃烧的Pythorch之后进行跟踪和可视化

torchember的Python项目详细描述


火炬余烬

Tracking and visualize after the burning pytorch

该框架跟踪Pythorch模型:

  • nn.Module级别
  • 下至所有张量的度量/特征,包括
    • 每个模块的输入/输出
    • 重量/梯度张量
  • 通过minimal额外编码

WebUI

其他可爱的特征

  • 可定制的指标,使用简单的装饰语法
  • 以您喜欢的方式拆分跟踪日志,只需mark(k=v,k1=v2...)
  • 您可以轻松打开/关闭跟踪:
    • 即使计算成本很小,torchember也不必为每次迭代计算度量
    • 因此,你可以跟踪例如,只有最后一步,只有每200步。等等

安装

pip install torchember

快速教程

第一步,跟踪你的模型

把你的火把余烬追踪器放在你的模型上

fromtorchember.coreimporttorchEmberte=torchEmber(model)

上面可以跟踪每个模块的输入和输出,下面可以跟踪每个模块的状态

^{pr2}$

像往常一样训练你的模特

步骤2,检查WebUI上的分析

从终端运行服务

$ torchember

默认端口为8080

或者分配一个端口

$ torchember --port=4200

请访问您的分析网站http://[host]:[port]

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
ws-consumer服务地址中的java动态属性   java如何比较整数列表,然后按升序排序?   javascript我正在使用java脚本调用一个函数,但它没有调用代码下面的方法,也没有调用secretitnames()函数   在文本窗格中多次使用Java insertIcon图标   JavaMSAL安卓。AuthenticationActivity完成,但用于身份验证请求的线程池线程仍处于等待状态   if语句中的java多范围比较   java toString()表示输出   java如何在jcstrest测试中生成指令重新排序   java我怎样才能运行它?   web应用程序中使用Hibernate和Spring的java问题   如何将字符串数据写入Java文本文件   如何在java命令提示符下运行已签名的jar文件?   java从我的菜单调用RCP应用程序   java如何等待Canvas/GraphicsContext完成任务,然后再继续执行代码块?