https://github.com/raynardj/drtensor
Debug tool to help pytorch modeling
https://github.com/raynardj/drtensor
Last synced: about 1 year ago
JSON representation
Debug tool to help pytorch modeling
- Host: GitHub
- URL: https://github.com/raynardj/drtensor
- Owner: raynardj
- License: gpl-3.0
- Created: 2024-04-18T16:50:34.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2024-04-19T05:24:46.000Z (about 2 years ago)
- Last Synced: 2025-03-05T05:46:49.419Z (over 1 year ago)
- Language: Python
- Size: 25.4 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# 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.
```bash
pip install drtensor
```
## 🚀 Usage
### With clause
```python
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:])
```