enochkan / torch-metrics
Metrics for model evaluation in pytorch
☆113Updated 4 years ago
Alternatives and similar repositories for torch-metrics:
Users that are interested in torch-metrics are comparing it to the libraries listed below
- Provides everything needed for high performance data loading and augmentation in pytorch.☆319Updated last year
- Semantic Segmentation with Pytorch-Lightning☆63Updated 4 years ago
- A pytorch dataset sampler for always sampling balanced batches.☆114Updated 4 years ago
- This is the code base for paper "G-SimCLR : Self-Supervised Contrastive Learning with Guided Projection via Pseudo Labelling" by Souradip…☆77Updated 3 years ago
- A collection of code snippets for my PyTorch Lightning projects☆107Updated 4 years ago
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆133Updated 4 years ago
- AutoML for image augmentation. AutoAlbument uses the Faster AutoAugment algorithm to find optimal augmentation policies. Documentation - …☆205Updated 3 years ago
- Official PyTorch code for U-Noise: Learnable Noise Masks for Interpretable Image Segmentation (ICIP 2021)☆41Updated 3 years ago
- Tool box for PyTorch☆187Updated last year
- ☆92Updated 4 years ago
- Implements the ideas presented in https://arxiv.org/pdf/2004.11362v1.pdf by Khosla et al.☆132Updated 4 years ago
- Unofficial PyTorch Implementation of EvoNorm☆121Updated 3 years ago
- Repo to build on / reproduce the record breaking Ranger-Mish-SelfAttention setup on FastAI ImageWoof dataset 5 epochs☆116Updated 5 years ago
- NeuralPy: A Keras like deep learning library works on top of PyTorch☆78Updated 9 months ago
- 🎲 Iterable dataset resampling in PyTorch☆91Updated 3 years ago
- A lightweight module for Multi-Task Learning in pytorch.☆160Updated last year
- An active learning library for Pytorch based on Lightning-Fabric.☆79Updated 10 months ago
- A regular Tiramisu network for PyTorch with a LOT of extra features!☆42Updated 3 years ago
- A simpler version of the self-attention layer from SAGAN, and some image classification results.☆212Updated 5 years ago
- ☆161Updated 2 years ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 2 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆329Updated 2 years ago
- (Minimally) implements SimCLR (https://arxiv.org/abs/2002.05709) in TensorFlow 2.☆90Updated 3 years ago
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends☆77Updated 3 months ago
- The official repository for "Intermediate Layers Matter in Momentum Contrastive Self Supervised Learning" paper.☆40Updated 2 years ago
- TF 2.x and PyTorch Lightning Callbacks for GPU monitoring☆92Updated 4 years ago
- Implementation of self-supervised image-level contrastive pretraining methods using Keras.☆70Updated 3 years ago
- Pytorch notebook with One Cycle Policy implementation (https://arxiv.org/abs/1803.09820)☆74Updated 2 years ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- Keras implementation of Normalizer-Free Networks and SGD - Adaptive Gradient Clipping☆70Updated 3 years ago