SkylerHuang

waymo_to_argoverse

Waymo Open Dataset -> Argoverse Converter

pytorch_geometric_temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

transformer

A Pytorch Implementation of "Attention is All You Need" and "Weighted Transformer Network for Machine Translation"

HGNN

Hypergraph Neural Networks (AAAI 2019)

SGCN

Code for "SGCN:Sparse Graph Convolution Network for Pedestrian Trajectory Prediction" CVPR 2021

hummingbird

Hummingbird compiles trained ML models into tensor computation for faster inference.

LaneGCN

[ECCV2020 Oral] Learning Lane Graph Representations for Motion Forecasting

MobileBert_PyTorch

MobileBERT: a Compact Task-Agnostic BERT for Resource-Limited Devices

TinyBert

简洁易用版TinyBert：基于Bert进行知识蒸馏的预训练语言模型

awesome-fast-attention

list of efficient attention modules

awesome-transformer-nlp

A curated list of NLP resources focused on Transformer networks, attention mechanism, GPT, BERT, ChatGPT, LLMs, and transfer learning.

pytorch-original-transformer

My implementation of the original transformer model (Vaswani et al.). I've additionally included the playground.py file for visualizing otherwise seemingly hard concepts. Currently included IWSLT pretrained models.

darts

Differentiable architecture search for convolutional and recurrent networks

attention-is-all-you-need-pytorch

A PyTorch implementation of the Transformer model in "Attention is All You Need".

awesome-bert

bert nlp papers, applications and github resources, including the newst xlnet ， BERT、XLNet 相关论文和 github 项目

DNA

(CVPR 2020) Block-wisely Supervised Neural Architecture Search with Knowledge Distillation

FNA

Fast Neural Network Adaptation via Parameter Remapping and Architecture Search (ICLR2020 & TPAMI)

Awesome-AutoDL

Automated Deep Learning: Neural Architecture Search Is Not the End (a curated list of AutoDL resources and an in-depth analysis)

soft-filter-pruning

Soft Filter Pruning for Accelerating Deep Convolutional Neural Networks

network-slimming

Network Slimming (Pytorch) (ICCV 2017)

pytorch-slimming

Learning Efficient Convolutional Networks through Network Slimming, In ICCV 2017.

MetaPruning

MetaPruning: Meta Learning for Automatic Neural Network Channel Pruning. In ICCV 2019.

Awesome-Pruning

A curated list of neural network pruning resources.

knowledge-distillation-pytorch

A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility

rethinking-network-pruning

Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)

Relabeled-HELEN-Dataset

rectified version of the original HELEN dataset.

cv_note

记录cv算法工程师的成长之路，分享计算机视觉和模型压缩部署技术栈笔记。https://harleyszhang.github.io/cv_note/

ssl_3d_recon

Code release for "From Image Collections to Point Clouds with Self-supervised Shape and Pose Networks" (CVPR 2020)

Domain-Adaptive-Single-View-3D-Reconstruction

graphx-conv

Official implementation of GraphX-Convolution