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)
Interactively inspect module inputs, outputs, parameters, and gradients.
(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.
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 wi…
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
Unified Implementations of Offline Reinforcement Learning Algorithms
Practical Computer Vision Bootcamp with notebooks, quizzes, and videos
Easiest way of fine-tuning HuggingFace video classification models
Open-source PyTorch tool that traces live training signals back to the data samples causing them. Pause the training to find mislabels, outliers, and class imbalance in image, video, and LiDAR datasets. And train again.
Train Scene Graph Generation for Visual Genome and GQA in PyTorch >= 1.2 with improved zero and few-shot generalization [BMVC 2020, ICCV 2021]
Deep Learning project template for PyTorch (multi-gpu training is supported)
Add a description, image, and links to the wandb topic page so that developers can more easily learn about it.
To associate your repository with the wandb topic, visit your repo's landing page and select "manage topics."