ropas/pytea
PyTea: PyTorch Tensor shape error analyzer
This tool helps deep learning developers find and fix shape-related errors in their PyTorch code. It takes a Python script using PyTorch as input and analyzes potential issues with tensor dimensions and API usage. The output highlights where shape mismatches or incorrect API arguments might occur, helping developers avoid runtime errors.
323 stars. No commits in the last 6 months.
Use this if you are a PyTorch developer struggling with unexpected tensor shape errors or API misuses in your deep learning models.
Not ideal if your project uses deep learning frameworks other than PyTorch or if you are not a developer working with Python code.
Stars
323
Forks
17
Language
TypeScript
License
—
Category
Last pushed
Apr 26, 2022
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/ropas/pytea"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
pytorch/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras-team/keras
Deep Learning for humans
Lightning-AI/torchmetrics
Machine learning metrics for distributed, scalable PyTorch applications.
Lightning-AI/pytorch-lightning
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
lanpa/tensorboardX
tensorboard for pytorch (and chainer, mxnet, numpy, ...)