Tracking and Visualize after the burning PyTorch
Project description
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
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
-
30 seconds tutorial
-
Full documentations
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]
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
torchember-0.2.5.tar.gz
(1.8 MB
view hashes)
Built Distribution
Close
Hashes for torchember-0.2.5-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 77cc2c02b89719cfdd064ac263b9d9b4de3d77e263c75c57a1032ea5c0ba6479 |
|
MD5 | 37e8c7c9a29d19fcf6df99797302ee19 |
|
BLAKE2b-256 | 7dec2e137281c2c5e2d0e782c6c2f71b1de204083050069a2e220a913f082c5d |