Pytorch modeling easier
Project description
Doctor Tensor 🩺
pip install drtensor
Debug tool to help pytorch modeling
📦 Installation
We assume you have the pytorch already installed. Or this library doesn't serve any purpose.
pip install drtensor
🚀 Usage
With clause
from drtensor.doctor import DrTensor
model = reset_model()
dr_tensor = DrTensor(resnet=model)
with dr_tensor:
y_ = model(torch.randn(1, 3, 224, 224))
# check the logs
print(dr_tensor.logs[-5:])
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
drtensor-0.0.1.tar.gz
(15.9 kB
view details)
Built Distribution
drtensor-0.0.1-py3-none-any.whl
(16.8 kB
view details)
File details
Details for the file drtensor-0.0.1.tar.gz.
File metadata
- Download URL: drtensor-0.0.1.tar.gz
- Upload date:
- Size: 15.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.9.19
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 5ecf223cd001fd96ee9f11c83bf87c56f415101c7c19ef6bfbdc2adc5e466c75 |
|
| MD5 | ab943cc688b4ca6338c6e16cba97105f |
|
| BLAKE2b-256 | 058988c41a80dfbe6f8f4355592bf44cdd98a231cd59f09097f5a850914f89c5 |
Provenance
File details
Details for the file drtensor-0.0.1-py3-none-any.whl.
File metadata
- Download URL: drtensor-0.0.1-py3-none-any.whl
- Upload date:
- Size: 16.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.9.19
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | e090a08a348541fbea949a9846f59faecb3d45bf6ba7c41ed0f528f57bdc1453 |
|
| MD5 | 0c4beb57d06290ee1b0a8008ae64732d |
|
| BLAKE2b-256 | 58dd98d928d6c762aa910d7cbe53e6a162e6350762b9fbf3b061140f4055b452 |
