enochkan / torch-metrics

Related projects ⓘ

Alternatives and complementary repositories for torch-metrics

• ankandrew / online-label-smoothing-pt
  Implementation of Online Label Smoothing in PyTorch
  ☆94Updated 2 years ago
• PhoenixDL / rising
  Provides everything needed for high performance data loading and augmentation in pytorch.
  ☆316Updated 10 months ago
• borisdayma / lightning-kitti
  Semantic Segmentation with Pytorch-Lightning
  ☆63Updated 3 years ago
• galatolofederico / pytorch-balanced-batch
  A pytorch dataset sampler for always sampling balanced batches.
  ☆110Updated 3 years ago
• beresandras / contrastive-classification-keras
  Implementation of self-supervised image-level contrastive pretraining methods using Keras.
  ☆69Updated 3 years ago
• awaelchli / pytorch-lightning-snippets
  A collection of code snippets for my PyTorch Lightning projects
  ☆107Updated 3 years ago
• sicara / gpumonitor
  TF 2.x and PyTorch Lightning Callbacks for GPU monitoring
  ☆92Updated 4 years ago
• MECLabTUDA / Super_Selfish
  ☆92Updated 3 years ago
• szymonmaszke / torchdatasets
  PyTorch dataset extended with map, cache etc. (tensorflow.data like)
  ☆328Updated 2 years ago
• digantamisra98 / EvoNorm
  Unofficial PyTorch Implementation of EvoNorm
  ☆121Updated 3 years ago
• theblackcat102 / H5Record
  Large dataset storage format for Pytorch
  ☆45Updated 3 years ago
• alexhock / object-detection-metrics
  Python code for analysing object detection metrics
  ☆31Updated 2 years ago
• ariG23498 / G-SimCLR
  This is the code base for paper "G-SimCLR : Self-Supervised Contrastive Learning with Guided Projection via Pseudo Labelling" by Souradip…
  ☆75Updated 2 years ago
• anguelos / tormentor
  Pytorch augmentation
  ☆119Updated 11 months ago
• bonlime / pytorch-tools
  Tool box for PyTorch
  ☆186Updated 8 months ago
• antoinebrl / torchextractor
  Feature extraction made simple with torchextractor
  ☆101Updated 3 years ago
• Randl / kmeans_selfsuper
  ☆54Updated 3 years ago
• lucidrains / contrastive-learner
  A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network
  ☆123Updated 3 years ago
• sayakpaul / Adaptive-Gradient-Clipping
  Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.
  ☆80Updated 3 years ago
• sdoria / SimpleSelfAttention
  A simpler version of the self-attention layer from SAGAN, and some image classification results.
  ☆211Updated 5 years ago
• pabloppp / pytorch-tools
  Useful PyTorch functions and modules that are not implemented in PyTorch by default
  ☆187Updated 6 months ago
• ayulockin / SwAV-TF
  TensorFlow implementation of "Unsupervised Learning of Visual Features by Contrasting Cluster Assignments".
  ☆85Updated 2 years ago
• sayakpaul / Supervised-Contrastive-Learning-in-TensorFlow-2
  Implements the ideas presented in https://arxiv.org/pdf/2004.11362v1.pdf by Khosla et al.
  ☆132Updated 4 years ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆72Updated 3 years ago
• lucidrains / nystrom-attention
  Implementation of Nyström Self-attention, from the paper Nyströmformer
  ☆122Updated 10 months ago
• aakashrkaku / intermdiate_layer_matter_ssl
  The official repository for "Intermediate Layers Matter in Momentum Contrastive Self Supervised Learning" paper.
  ☆40Updated 2 years ago
• MaxHalford / pytorch-resample
  🎲 Iterable dataset resampling in PyTorch
  ☆91Updated 2 years ago
• albumentations-team / autoalbument
  AutoML for image augmentation. AutoAlbument uses the Faster AutoAugment algorithm to find optimal augmentation policies. Documentation - …
  ☆203Updated 3 years ago
• sayakpaul / MLP-Mixer-CIFAR10
  Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.
  ☆54Updated 2 years ago