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.

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.

Transformers 3rd Edition

Interactively inspect module inputs, outputs, parameters, and gradients.

PyTorch Implementations For A Series Of Deep Learning-Based Recommendation Models

wandbot is a technical support bot for Weights & Biases' AI developer tools that can run in Discord, Slack, ChatGPT and Zendesk

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

Deep Learning project template for PyTorch (multi-gpu training is supported)

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

A multibranch neural net architecture. (PyTorch)

👦 Human head semantic segmentation