Bjarten / early-stopping-pytorchLinks
Early stopping for PyTorch
☆1,267Updated 10 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
Sorting:
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,316Updated 2 months ago
- A learning rate range test implementation in PyTorch☆983Updated 2 months ago
- A PyTorch Implementation of Focal Loss.☆987Updated 5 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,055Updated last year
- A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.☆908Updated last year
- Some custom dataset examples for PyTorch☆868Updated 5 years ago
- PyTorch implementation of some attentions for Deep Learning Researchers.☆544Updated 3 years ago
- A small package to create visualizations of PyTorch execution graphs☆3,425Updated 8 months ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆992Updated 11 months ago
- [ICML 2021, Long Talk] Delving into Deep Imbalanced Regression☆888Updated 3 years ago
- Source code for "On the Relationship between Self-Attention and Convolutional Layers"☆1,110Updated 2 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,248Updated 11 months ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆800Updated last year
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,203Updated last year
- Confusion Matrix in Python: plot a pretty confusion matrix (like Matlab) in python using seaborn and matplotlib☆535Updated 5 months ago
- Train AI models efficiently on medical images using any framework☆1,874Updated last year
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,848Updated last year
- mixup: Beyond Empirical Risk Minimization☆1,190Updated 3 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,139Updated last year
- Pytorch Lightning code guideline for conferences☆1,279Updated 2 years ago
- Unsupervised Data Augmentation (UDA)☆2,197Updated 4 years ago
- A multi-task learning example for the paper https://arxiv.org/abs/1705.07115☆867Updated 5 years ago
- List of useful data augmentation resources. You will find here some not common techniques, libraries, links to GitHub repos, papers, and …☆1,647Updated last year
- PyTorch Library for Active Learning to accompany Human-in-the-Loop Machine Learning book☆973Updated 2 years ago
- A simple way to calibrate your neural network.☆1,162Updated last month
- PyTorch re-implementation of Grad-CAM (+ vanilla/guided backpropagation, deconvnet, and occlusion sensitivity maps)☆799Updated 5 years ago
- scikit-learn cross validators for iterative stratification of multilabel data☆876Updated 11 months ago
- Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.☆1,210Updated 4 years ago
- Siamese and triplet networks with online pair/triplet mining in PyTorch☆3,152Updated 2 years ago
- An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.☆238Updated last year