Dilin Wang's repositories
Stein-Variational-Gradient-Descent
code for the paper "Stein Variational Gradient Descent (SVGD): A General Purpose Bayesian Inference Algorithm"
fast-energy-aware-splitting
Energy-Aware Neural Architecture Optimization with Fast Splitting Steepest Descent
matrix_svgd
Stein Variational Gradient Descent with Matrix-Valued Kernels
AdamP
AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BinaryNet.pytorch
Binarized Neural Network (BNN) for pytorch
Cutout
2.56%, 15.20%, 1.30% on CIFAR10, CIFAR100, and SVHN https://arxiv.org/abs/1708.04552
detr
End-to-End Object Detection with Transformers
face-recognition
Deep face recognition with Keras, Dlib and OpenCV
knowledge-distillation-pytorch
A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility
L0_regularization
Learning Sparse Neural Networks through L0 regularization
mmsegmentation
OpenMMLab Semantic Segmentation Toolbox and Benchmark.
neural-architecture-transfer
Neural Architecture Transfer (Arxiv'20), PyTorch Implementation
NLP-progress
Repository to track the progress in Natural Language Processing (NLP), including the datasets and the current state-of-the-art for the most common NLP tasks.
once-for-all
[ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment
pretrained-models.pytorch
Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
pytorch-cnn-visualizations
Pytorch implementation of convolutional neural network visualization techniques
pytorch-image-models
PyTorch image models, scripts, pretrained weights -- (SE)ResNet/ResNeXT, DPN, EfficientNet, MixNet, MobileNet-V3/V2, MNASNet, Single-Path NAS, FBNet, and more
pytorch-lightning
The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.
pytorch-randaugment
Unofficial PyTorch Reimplementation of RandAugment.
pytorch_ema
Tiny PyTorch library for maintaining a moving average of a collection of parameters.
ResNeSt
ResNeSt: Split-Attention Networks
sam
SAM: Sharpness-Aware Minimization (PyTorch)
stable-dreamfusion
A pytorch implementation of text-to-3D dreamfusion, powered by stable diffusion.
vision
Datasets, Transforms and Models specific to Computer Vision
word_cloud
A little word cloud generator in Python