Bjarten / early-stopping-pytorch
Early stopping for PyTorch
☆1,246Updated 4 months ago
Alternatives and similar repositories for early-stopping-pytorch:
Users that are interested in early-stopping-pytorch are comparing it to the libraries listed below
- A PyTorch Implementation of Focal Loss.☆979Updated 5 years ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,291Updated 2 months ago
- Some custom dataset examples for PyTorch☆865Updated 4 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,033Updated last year
- A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.☆892Updated last year
- A learning rate range test implementation in PyTorch☆950Updated 3 months ago
- PyTorch re-implementation of Grad-CAM (+ vanilla/guided backpropagation, deconvnet, and occlusion sensitivity maps)☆796Updated 5 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,224Updated 5 months ago
- Model interpretability and understanding for PyTorch☆5,136Updated this week
- torch-optimizer -- collection of optimizers for Pytorch☆3,092Updated last year
- Train AI models efficiently on medical images using any framework☆1,872Updated 9 months ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆988Updated 5 months ago
- A curated list of resources for Learning with Noisy Labels☆2,665Updated 10 months ago
- mixup: Beyond Empirical Risk Minimization☆1,177Updated 3 years ago
- Source code for "On the Relationship between Self-Attention and Convolutional Layers"☆1,098Updated 2 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆792Updated last year
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,197Updated last year
- Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"☆2,119Updated 2 years ago
- Pytorch Lightning code guideline for conferences☆1,260Updated last year
- View model summaries in PyTorch!☆2,730Updated last week
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,818Updated last year
- Code for visualizing the loss landscape of neural nets☆2,938Updated 2 years ago
- Stochastic Weight Averaging in PyTorch☆969Updated 3 years ago
- A small package to create visualizations of PyTorch execution graphs☆3,320Updated 2 months ago
- PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)☆3,220Updated last year
- PyTorch implementation of Contrastive Learning methods☆1,966Updated last year
- On the Variance of the Adaptive Learning Rate and Beyond☆2,546Updated 3 years ago
- An unofficial styleguide and best practices summary for PyTorch☆1,966Updated 3 years ago
- PyTorch deep learning projects made easy.☆4,880Updated 9 months ago
- A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility☆1,921Updated 2 years ago