Bjarten / early-stopping-pytorch
Early stopping for PyTorch
☆1,252Updated 5 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.☆983Updated 5 years ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,297Updated 2 weeks ago
- A learning rate range test implementation in PyTorch☆954Updated 4 months ago
- Some custom dataset examples for PyTorch☆866Updated 5 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,038Updated last year
- mixup: Beyond Empirical Risk Minimization☆1,178Updated 3 years ago
- A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.☆899Updated last year
- Pytorch Lightning code guideline for conferences☆1,261Updated last year
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆988Updated 6 months ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,233Updated 6 months ago
- A small package to create visualizations of PyTorch execution graphs☆3,342Updated 3 months ago
- Source code for "On the Relationship between Self-Attention and Convolutional Layers"☆1,100Updated 2 years ago
- PyTorch implementation of some attentions for Deep Learning Researchers.☆531Updated 3 years ago
- PyTorch re-implementation of Grad-CAM (+ vanilla/guided backpropagation, deconvnet, and occlusion sensitivity maps)☆796Updated 5 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,103Updated last year
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,825Updated last year
- Train AI models efficiently on medical images using any framework☆1,874Updated 10 months ago
- Learning Rate Warmup in PyTorch☆409Updated last month
- Model interpretability and understanding for PyTorch☆5,181Updated 2 weeks ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,548Updated 3 years ago
- PyTorch deep learning projects made easy.☆4,905Updated 10 months ago
- list of efficient attention modules☆1,000Updated 3 years ago
- A lightweight library for PyTorch training tools and utilities☆1,691Updated 2 weeks ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆796Updated last year
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,199Updated last year
- PyTorch Boilerplate For Research☆609Updated 3 years ago
- Implementation of gMLP, an all-MLP replacement for Transformers, in Pytorch☆425Updated 3 years ago
- Implementation of Convolutional LSTM in PyTorch.☆2,102Updated 4 years ago
- Bayesian Convolutional Neural Network with Variational Inference based on Bayes by Backprop in PyTorch.☆1,495Updated last year
- Reformer, the efficient Transformer, in Pytorch☆2,163Updated last year