dilinwang820

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

dilinwang.github.io

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

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

MABN

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.

RAFT-Stereo

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