ranamihir/pytorch_common
A lightweight wrapper that contains PyTorch code that is common to most projects built on PyTorch.
This is a tool for machine learning engineers and researchers who are building deep learning models using PyTorch. It provides a streamlined way to train, test, and manage models, handling common tasks like data loading, model initialization, logging, and checkpointing. You provide your data and model architecture, and it helps manage the training process and output model predictions or evaluation metrics.
No commits in the last 6 months.
Use this if you are a machine learning engineer or researcher building PyTorch models and want a flexible, pre-optimized structure for common deep learning workflows, especially when managing configuration settings through YAML files.
Not ideal if you are looking for a high-level, no-code solution for machine learning or if you prefer to build every aspect of your PyTorch training loop from scratch without any abstractions.
Stars
7
Forks
1
Language
Python
License
MIT
Category
Last pushed
Mar 24, 2022
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/ranamihir/pytorch_common"
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, ...)