Skip to main content

Torch Ember

Tracking and visualize after the burning pytorch

This framework tracks the pytorch model:

  • On nn.Module level
  • Down to the metrics/ features of all tensors, includes
    • inputs/outputs of each module
    • weight/grad tensors
  • By minimal extra coding

WebUI

Other lovely features

  • Customizable metrics, with easy decorator syntax
  • Split the tracking log in the way you like, just mark(k=v,k1=v2...)
  • You can easily switch on/off the tracking:
    • Even cost of computation is tiny, torchember don't have to calculate metric for every iteration
    • Hence, you can track eg. only the last steps, only each 200 steps .etc

Installation

pip install torchember

Fast Tutorial

Step1, Track your model

Place you torch ember tracker on your model

from torchember.core import torchEmber
te = torchEmber(model)

The above can track input and output of every module,The following can track status of every module

for i in range(1000):
    ...
    loss.backward()
    optimizer.step()

    te.log_model()

Train your model as usual

Step2, Check the analysis on the WebUI

Run the service from terminal

$ torchember

The default port will be 8080

Or assign a port

$ torchember --port=4200

Visit your analysis at http://[host]:[port]

Metadata

Release files for torchember 0.2.7

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for torchember 0.2.7
File Size Uploaded Provenance
torchember-0.2.7.tar.gz 1.8 MB Without provenance Details

Built distribution (wheel)

Table of built distributions (wheels) for torchember 0.2.7
File Interpreter ABI Platform
torchember-0.2.7-py3-none-any.whl Python 3 none any Details

Total release size: 3.7 MB

Release files / torchember-0.2.7.tar.gz

Download URL torchember-0.2.7.tar.gz
Size 1.8 MB
Tags Source
SHA-256 checksum
How to use checksums
41bf4dc3b90deeb20a3d29d9373f303c4a7e679f50d8f04d9ad0d820c120435e
BLAKE2b-256 checksum
How to use checksums
7df46433945cbbb30091bc0b37ade09e24138eaa0dd47c1937be8c4d4b71ec78
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/3.1.1 pkginfo/1.5.0.1 requests/2.22.0 setuptools/45.1.0 requests-toolbelt/0.9.1 tqdm/4.42.1 CPython/3.7.4

Release files / torchember-0.2.7-py3-none-any.whl

Download URL torchember-0.2.7-py3-none-any.whl
Size 1.9 MB
Tags Python 3
SHA-256 checksum
How to use checksums
f90c46d0d2697367e72858c83e206f85a109b44ad7614a2da4637ed8d7fed993
BLAKE2b-256 checksum
How to use checksums
832a2680adec8ceacdb68ab05639b045903baa7b191ab87b87944ec538268053
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/3.1.1 pkginfo/1.5.0.1 requests/2.22.0 setuptools/45.1.0 requests-toolbelt/0.9.1 tqdm/4.42.1 CPython/3.7.4

Security for torchember 0.2.7

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page