A PyTorch re-implementation of Mask Autoencoder training. SimpleMAE tries to be small, clean, interpretable and extensible.
pip install -r requirements.txt
(base) ➜ python main.py -c configs/imagenet.yaml --debug
Outputs will be saved to ../mae_out/imagenet-1116211657
Found 0 gpu(s)
Param num: 76445440
Epoch 0 Train loss 1.5653417110443115
Output has been saved to ../mae_out/imagenet-1116211657
PatrickStar: Parallel Training of Large Language Models via a Chunk-based Memory Management Meeting PatrickStar Pre-Trained Models (PTM) are becoming
Face-Transformer This is the code of Face Transformer for Recognition (https://arxiv.org/abs/2103.14803v2). Recently there has been great interests of
NLopt is a library for nonlinear local and global optimization, for functions with and without gradient information. It is designed as a simple, unifi
Real-ESRGAN Colab Demo for Real-ESRGAN . Portable Windows executable file. You can find more information here. Real-ESRGAN aims at developing Practica
pytorch-unflow This is a personal reimplementation of UnFlow [1] using PyTorch. Should you be making use of this work, please cite the paper according
Dual-Resolution Correspondence Network Dual-Resolution Correspondence Network, NeurIPS 2020 Dependency All dependencies are included in asset/dualrcne
Spiking neural network (SNN), compared with depth neural network (DNN), has faster processing speed, lower energy consumption and more biological interpretability, which is expected to approach Stron
SiamTracker-with-TensorRT The modify PyTorch version of Siam-trackers which are speed-up by TensorRT or ONNX. [Updating...] Examples demonstrating how
Scene-Text-Detection-and-Recognition (Pytorch) Competition URL: https://tbrain.t
This is a simple Python game for finding a heart emoji. There is a 3 x 3 matrix in which a heart emoji resides. The location of the heart is randomized and is not revealed. The player must guess the
The Breathing K-Means Algorithm (with examples) The Breathing K-Means is an approximation algorithm for the k-means problem that (on average) is bette
SUCP Leveraging Social Influence based on Users Activity Centers for Point-of-Interest Recommendation () Direct Friends (i.e., users who follow each o
Fibonaaci Series Generating the Fibonacci sequence is a classic recursive proble
MosaicKD Code for NeurIPS-21 paper "Mosaicking to Distill: Knowledge Distillation from Out-of-Domain Data" 1. Motivation Natural images share common l
PERSIA (Parallel rEcommendation tRaining System with hybrId Acceleration) is developed by AI
FaceX-Zoo FaceX-Zoo is a PyTorch toolbox for face recognition. It provides a training module with various supervisory heads and backbones towards stat
Frequency Spectrum Augmentation Consistency for Domain Adaptive Object Detection Main requirements torch = 1.0 torchvision = 0.2.0 Python 3 Environm
Exposure: A White-Box Photo Post-Processing Framework ACM Transactions on Graphics (presented at SIGGRAPH 2018) Yuanming Hu1,2, Hao He1,2, Chenxi Xu1,
Status: Under development (expect bug fixes and huge updates) ShinRL: A Library for Evaluating RL Algorithms from Theoretical and Practical Perspectiv
Generative Models Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow. Also present here are RBM and Helmholtz Machine. Note: Gen
633 Dec 28, 2022
153 Nov 30, 2022
1.4k Dec 25, 2022
17.2k Jan 02, 2023
134 Nov 20, 2022
45 Nov 21, 2022
18 Dec 09, 2022
9 Dec 13, 2022
9 Jan 02, 2023
1 Jan 24, 2022
75 Nov 17, 2022
8 Nov 26, 2022
1 Feb 15, 2022
37 Nov 10, 2022
1.6k Jan 06, 2023
15 Apr 04, 2022
719 Dec 29, 2022
37 Dec 28, 2022
7k Jan 02, 2023