frank-xwang

XuDong Frank Wang's starred repositories

DALL-E

PyTorch package for the discrete VAE used for DALL·E.

Language:PythonNOASSERTION10765 232 89

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Language:PythonBSD-3-Clause8207 102 1165

dino

PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO

Language:PythonApache-2.06107 68 246

fourier-feature-networks

Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains

Language:Jupyter NotebookMIT1206 24 20

convmixer

Implementation of ConvMixer for "Patches Are All You Need? 🤷"

Language:PythonMIT1052 14 16

FixMatch-pytorch

Unofficial PyTorch implementation of "FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidence"

Language:PythonMIT747 9 53

Code to reproduce the results in the FAIR research papers "Semi-Supervised Learning of Visual Features by Non-Parametrically Predicting View Assignments with Support Samples" https://arxiv.org/abs/2104.13963 and "Supervision Accelerates Pre-training in Contrastive Semi-Supervised Learning of Visual Representations" https://arxiv.org/abs/2006.10803

Language:PythonMIT487 13 30

RIDE-LongTailRecognition

[ICLR 2021 Spotlight] Code release for "Long-tailed Recognition by Routing Diverse Distribution-Aware Experts."

Language:PythonMIT257 6 32

exponential-moving-average-normalization

PyTorch implementation of EMAN for self-supervised and semi-supervised learning: https://arxiv.org/abs/2101.08482

Language:Python98 11 4

CLD-UnsupervisedLearning

[CVPR 2021] Code release for "Unsupervised Feature Learning by Cross-Level Instance-Group Discrimination."

Language:PythonMIT98 8 23

towards-universal-object-detection

[CVPR 2019] Pytorch implementation for "Towards Universal Object Detection by Domain Attention."

Language:Python75 2 5

TBC-TiedBlockConvolution

[AAAI 2021] Pytorch implementation for "Tied Block Convolution: Leaner and Better CNNs with Shared Thinner Filters."

Language:Python31 2 3

VAI-ReinforcementLearning

[CVPR 2021] Official Implementation of VAI: Unsupervised Visual Attention and Invariance for Reinforcement Learning

Language:Python25 4 8

VAI-ReinforcementLearning

[CVPR 2021] Pytorch implementation for "Unsupervised Visual Attention and Invariance for Reinforcement Learning."

4 10