Repository navigation
wandb
- Website
- Wikipedia
PyTorch implementation of the U-Net for image semantic segmentation with high quality images
High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
Github repo with tutorials to fine tune transformers for diff NLP tasks
Train a state-of-the-art yolov3 object detector from scratch!
Generic template to bootstrap your PyTorch project.
Transformers 3rd Edition
Unofficial PyTorch Implementation for HifiFace (https://arxiv.org/abs/2106.09965)
(ICML 2020) This repo contains code for our paper "Revisiting Training Strategies and Generalization Performance in Deep Metric Learning" (https://arxiv.org/abs/2002.08473) to facilitate consistent research in the field of Deep Metric Learning.
Interactively inspect module inputs, outputs, parameters, and gradients.
PyTorch Implementations For A Series Of Deep Learning-Based Recommendation Models
Phoneme Recognition using pre-trained models Wav2vec2, HuBERT and WavLM. Throughout this project, we compared specifically three different self-supervised models, Wav2vec (2019, 2020), HuBERT (2021) and WavLM (2022) pretrained on a corpus of English speech that we will use in various ways to perform phoneme recognition for different languages with a network trained with Connectionist Temporal Classification (CTC) algorithm.
Flexible and scalable template based on PyTorch Lightning + Hydra. Efficient workflow and reproducibility for rapid ML experiments.
KUIELAB-MDX-Net got the 2nd place on the Leaderboard A and the 3rd place on the Leaderboard B in the MDX-Challenge ISMIR 2021
Easiest way of fine-tuning HuggingFace video classification models
Train Scene Graph Generation for Visual Genome and GQA in PyTorch >= 1.2 with improved zero and few-shot generalization.
Tensorflow 2.3.0 implementation of DeepLabV3-Plus
👦 Human head semantic segmentation