Bjarten / early-stopping-pytorch

Related projects ⓘ

Alternatives and complementary repositories for early-stopping-pytorch

• ufoym / imbalanced-dataset-sampler
  A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.
  ☆2,276Updated last month
• davidtvs / pytorch-lr-finder
  A learning rate range test implementation in PyTorch
  ☆925Updated 2 months ago
• sksq96 / pytorch-summary
  Model summary in PyTorch similar to `model.summary()` in Keras
  ☆4,020Updated 8 months ago
• utkuozbulak / pytorch-custom-dataset-examples
  Some custom dataset examples for PyTorch
  ☆862Updated 4 years ago
• szagoruyko / pytorchviz
  A small package to create visualizations of PyTorch execution graphs
  ☆3,236Updated 7 months ago
• clcarwin / focal_loss_pytorch
  A PyTorch Implementation of Focal Loss.
  ☆963Updated 5 years ago
• waleedka / hiddenlayer
  Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.
  ☆1,800Updated 9 months ago
• moemen95 / Pytorch-Project-Template
  A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.
  ☆872Updated last year
• epfml / attention-cnn
  Source code for "On the Relationship between Self-Attention and Convolutional Layers"
  ☆1,086Updated last year
• pytorch / captum
  Model interpretability and understanding for PyTorch
  ☆4,939Updated this week
• paperswithcode / releasing-research-code
  Tips for releasing research code in Machine Learning (with official NeurIPS 2020 recommendations)
  ☆2,587Updated last year
• nitrain / nitrain
  Train AI models efficiently on medical images using any framework
  ☆1,866Updated 5 months ago
• ildoonet / pytorch-gradual-warmup-lr
  Gradually-Warmup Learning Rate Scheduler for PyTorch
  ☆977Updated last month
• TylerYep / torchinfo
  View model summaries in PyTorch!
  ☆2,604Updated this week
• gpleiss / temperature_scaling
  A simple way to calibrate your neural network.
  ☆1,104Updated 3 years ago
• CoinCheung / pytorch-loss
  label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful
  ☆2,185Updated last month
• lessw2020 / Ranger-Deep-Learning-Optimizer
  Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase
  ☆1,194Updated 11 months ago
• jettify / pytorch-optimizer
  torch-optimizer -- collection of optimizers for Pytorch
  ☆3,044Updated 8 months ago
• Lightning-AI / deep-learning-project-template
  Pytorch Lightning code guideline for conferences
  ☆1,240Updated last year
• AntixK / PyTorch-VAE
  A Collection of Variational Autoencoders (VAE) in PyTorch.
  ☆6,697Updated 5 months ago
• tomgoldstein / loss-landscape
  Code for visualizing the loss landscape of neural nets
  ☆2,840Updated 2 years ago
• YyzHarry / imbalanced-regression
  [ICML 2021, Long Talk] Delving into Deep Imbalanced Regression
  ☆826Updated 2 years ago
• victoresque / pytorch-template
  PyTorch deep learning projects made easy.
  ☆4,762Updated 5 months ago
• Lightning-Universe / lightning-bolts
  Toolbox of models, callbacks, and datasets for AI/ML researchers.
  ☆1,696Updated 2 weeks ago
• yaringal / multi-task-learning-example
  A multi-task learning example for the paper https://arxiv.org/abs/1705.07115
  ☆843Updated 4 years ago
• HobbitLong / PyContrast
  PyTorch implementation of Contrastive Learning methods
  ☆1,943Updated last year
• lucidrains / reformer-pytorch
  Reformer, the efficient Transformer, in Pytorch
  ☆2,123Updated last year
• Lightning-AI / torchmetrics
  Machine learning metrics for distributed, scalable PyTorch applications.
  ☆2,141Updated this week
• trent-b / iterative-stratification
  scikit-learn cross validators for iterative stratification of multilabel data
  ☆852Updated last month
• vandit15 / Class-balanced-loss-pytorch
  Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"
  ☆785Updated 9 months ago