Bjarten / early-stopping-pytorchView external linksLinks
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,878Jun 13, 2024Updated last year
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,731Feb 9, 2026Updated last week
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,064Mar 2, 2024Updated last year
- A scikit-learn compatible neural network library that wraps PyTorch☆6,149Dec 22, 2025Updated last month
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,351Updated this week
- Graph Neural Network Library for PyTorch☆23,469Updated this week
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆23,740Sep 1, 2025Updated 5 months ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,327Jan 5, 2026Updated last month
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,823Feb 4, 2026Updated last week
- A PyTorch implementation of the Transformer model in "Attention is All You Need".☆9,625Apr 16, 2024Updated last year
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆12,608Apr 7, 2025Updated 10 months ago
- Model interpretability and understanding for PyTorch☆5,556Feb 3, 2026Updated last week
- PyTorch implementations of Generative Adversarial Networks.☆17,423Jun 18, 2024Updated last year
- A learning rate range test implementation in PyTorch☆997Jun 24, 2025Updated 7 months ago
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆15,269Jun 25, 2025Updated 7 months ago
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…☆24,993Updated this week
- A PyTorch implementation of EfficientNet☆8,220Apr 8, 2022Updated 3 years ago
- Latex code for making neural networks diagrams☆24,422Aug 21, 2023Updated 2 years ago
- Graph Convolutional Networks in PyTorch☆5,397Sep 20, 2020Updated 5 years ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,987Feb 5, 2026Updated last week
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,304Aug 17, 2025Updated 5 months ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,161Mar 22, 2024Updated last year
- Python package built to ease deep learning on graph, on top of existing DL frameworks.☆14,229Jul 31, 2025Updated 6 months ago
- 🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal model…☆156,440Updated this week
- PyTorch Tutorial for Deep Learning Researchers☆32,177Aug 15, 2023Updated 2 years ago
- Datasets, Transforms and Models specific to Computer Vision☆17,510Updated this week
- Tutorials on implementing a few sequence-to-sequence (seq2seq) models with PyTorch and TorchText.☆5,675Jan 20, 2024Updated 2 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,116Apr 22, 2022Updated 3 years ago
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,406Jan 30, 2026Updated 2 weeks ago
- Must-read papers on graph neural networks (GNN)☆16,723Dec 20, 2023Updated 2 years ago
- A comprehensive list of pytorch related content on github,such as different models,implementations,helper libraries,tutorials etc.☆16,376Feb 1, 2026Updated 2 weeks ago
- State-of-the-Art Text Embeddings☆18,225Feb 9, 2026Updated last week
- A Collection of Variational Autoencoders (VAE) in PyTorch.☆7,587Mar 21, 2025Updated 10 months ago
- PyTorch deep learning projects made easy.☆5,091Jun 4, 2024Updated last year
- Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.☆11,324Dec 23, 2025Updated last month
- PyTorch implementation of the U-Net for image semantic segmentation with high quality images☆11,189Aug 11, 2024Updated last year
- Sequence modeling benchmarks and temporal convolutional networks☆4,474Mar 28, 2022Updated 3 years ago
- PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations☆2,484Mar 4, 2024Updated last year
- Basic Utilities for PyTorch Natural Language Processing (NLP)☆2,226Jul 4, 2023Updated 2 years ago