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 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
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."