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 hashes)
Built Distribution
drtensor-0.0.2-py3-none-any.whl
(17.9 kB
view hashes)