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 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)
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.1.7.tar.gz
(1.8 MB
view hashes)
Built Distribution
Close
Hashes for torchember-0.1.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 262f78ece2127d7bd35f5bb1e130b15bfb9c664345a650e0f5b3fb55a6aabc6a |
|
MD5 | ce9f95e53936e414ec6d451b43d6931e |
|
BLAKE2b-256 | c09ed0decfa0b551f677fc27360891cd1a48bdf2e258b3dc9c317c05c699f72f |