Stars
PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)
Adaptive FNO transformer - official Pytorch implementation
[NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification
[TMLR] "SC2 Benchmark: Supervised Compression for Split Computing"
Official code for Conformer: Local Features Coupling Global Representations for Visual Recognition
[NeurIPS 2019] This is the code repo of our novel passport-based DNN ownership verification schemes, i.e. we embed passport layer into various deep learning architectures (e.g. AlexNet, ResNet) for…
PyTorch implementations of Generative Adversarial Networks.
LPIPS metric. pip install lpips
This is the official implementation of NNSplitter (ICML'23)
A Collection of Variational Autoencoders (VAE) in PyTorch.
This is a term project for ELE851 - Detection & Estimation Theory - Spring 2021
[NeurIPS 2022] DataMUX: Data Multiplexing for Neural Networks
Code for paper "Orthogonal Convolutional Neural Networks".
《代码随想录》LeetCode 刷题攻略：200道经典题目刷题顺序，共60w字的详细图解，视频难点剖析，50余张思维导图，支持C++，Java，Python，Go，JavaScript等多语言版本，从此算法学习不再迷茫！🔥🔥 来看看，你会发现相见恨晚！🚀
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Loss Functions in the Era of Semantic Segmentation: A Survey and Outlook
Pretrained models on CIFAR10/100 in PyTorch
Code for AISTATS 2023 paper - Estimating Total Correlation with Mutual Information Estimators
Code for ICML2020 paper - CLUB: A Contrastive Log-ratio Upper Bound of Mutual Information
Tight Mutual Information Estimation With Contrastive Fenchel-Legendre Optimization
Official code for ICML 2022: Mitigating Neural Network Overconfidence with Logit Normalization