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:])
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
drtensor-0.0.2.tar.gz
(17.2 kB
view details)
Built Distribution
drtensor-0.0.2-py3-none-any.whl
(17.9 kB
view details)
File details
Details for the file drtensor-0.0.2.tar.gz
.
File metadata
- Download URL: drtensor-0.0.2.tar.gz
- Upload date:
- Size: 17.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.9.19
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 05abc489793757796b44da9eb8350569d6f7db0db7d98d667b516c2dd444fa06 |
|
MD5 | a183fe9a1ff3ba31f96325947579301e |
|
BLAKE2b-256 | ce38bdad9e319be79071ca2a844392cd4912fb4ee787187605f210a4e7da6afb |
File details
Details for the file drtensor-0.0.2-py3-none-any.whl
.
File metadata
- Download URL: drtensor-0.0.2-py3-none-any.whl
- Upload date:
- Size: 17.9 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 | 0b25451766dd9d29bf8301261584db74ab1c9e725a583bfafbba33442884dc82 |
|
MD5 | 377d60b0940da35bac2dbad6dac8a6c9 |
|
BLAKE2b-256 | 8aee98ad27d5b4bd9b40c5a546de578418fe61645fcafbfce070813592fed0ca |