Early stopping for PyTorch
☆1,270Nov 11, 2024Updated last year
Alternatives and similar repositories for early-stopping-pytorch
Users that are interested in early-stopping-pytorch are comparing it to the libraries listed below
Sorting:
- Train AI models efficiently on medical images using any framework☆1,879Jun 13, 2024Updated last year
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,752Updated this week
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,064Mar 2, 2024Updated 2 years ago
- A scikit-learn compatible neural network library that wraps PyTorch☆6,150Feb 25, 2026Updated 2 weeks ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,504Updated this week
- Graph Neural Network Library for PyTorch☆23,565Updated this week
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆23,793Sep 1, 2025Updated 6 months ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,324Jan 5, 2026Updated 2 months ago
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,926Updated this week
- A PyTorch implementation of the Transformer model in "Attention is All You Need".☆9,657Apr 16, 2024Updated last year
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆12,682Apr 7, 2025Updated 11 months ago
- Model interpretability and understanding for PyTorch☆5,572Mar 10, 2026Updated last week
- PyTorch implementations of Generative Adversarial Networks.☆17,451Jun 18, 2024Updated last year
- A learning rate range test implementation in PyTorch☆1,003Jun 24, 2025Updated 8 months ago
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆15,275Jun 25, 2025Updated 8 months ago
- A PyTorch implementation of EfficientNet☆8,218Apr 8, 2022Updated 3 years ago
- Latex code for making neural networks diagrams☆24,487Aug 21, 2023Updated 2 years ago
- Graph Convolutional Networks in PyTorch☆5,399Sep 20, 2020Updated 5 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,989Feb 5, 2026Updated last month
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,311Aug 17, 2025Updated 6 months ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,166Mar 22, 2024Updated last year
- Python package built to ease deep learning on graph, on top of existing DL frameworks.☆14,245Jul 31, 2025Updated 7 months ago
- 🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal model…☆157,783Updated this week
- PyTorch Tutorial for Deep Learning Researchers☆32,220Aug 15, 2023Updated 2 years ago
- Datasets, Transforms and Models specific to Computer Vision☆17,558Updated this week
- Tutorials on implementing a few sequence-to-sequence (seq2seq) models with PyTorch and TorchText.☆5,679Jan 20, 2024Updated 2 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,112Apr 22, 2022Updated 3 years ago
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,418Mar 9, 2026Updated last week
- Must-read papers on graph neural networks (GNN)☆16,732Dec 20, 2023Updated 2 years ago
- A comprehensive list of pytorch related content on github,such as different models,implementations,helper libraries,tutorials etc.☆16,421Feb 1, 2026Updated last month
- State-of-the-Art Text Embeddings☆18,390Updated this week
- A Collection of Variational Autoencoders (VAE) in PyTorch.☆7,603Mar 21, 2025Updated 11 months ago
- PyTorch deep learning projects made easy.☆5,106Jun 4, 2024Updated last year
- Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.☆11,403Updated this week
- PyTorch implementation of the U-Net for image semantic segmentation with high quality images☆11,254Aug 11, 2024Updated last year
- Sequence modeling benchmarks and temporal convolutional networks☆4,484Mar 28, 2022Updated 3 years ago
- A game theoretic approach to explain the output of any machine learning model.☆25,106Mar 7, 2026Updated last week
- PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations☆2,479Mar 4, 2024Updated 2 years ago
- Basic Utilities for PyTorch Natural Language Processing (NLP)☆2,228Jul 4, 2023Updated 2 years ago