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
aws-documentation We present the AWS documentation corpus, an open-book QA dataset, which contains 25,175 documents along with 100 matched questions a
使用yolov5训练自己的数据集（详细过程）并通过flask部署 依赖库 torch torchvision numpy opencv-python lxml tqdm flask pillow tensorboard matplotlib pycocotools Windows，请使用 pycoc
LaneNet-Lane-Detection Use tensorflow to implement a Deep Neural Network for real time lane detection mainly based on the IEEE IV conference paper "To
Sparse Structure Learning via Graph Neural Networks for inductive document classification Make graph dataset create co-occurrence graph for datasets.
RefineGAN unofficial pytorch implementation of RefineGAN (https://arxiv.org/abs/1709.00753) for CSMRI reconstruction, the official code using tensorpa
Diffusion Normalizing Flow (DiffFlow) Reproduce setup environment The repo heavily depends on jam, a personal toolbox developed by Qsh.zh. The API may
In-Place Activated BatchNorm In-Place Activated BatchNorm for Memory-Optimized Training of DNNs In-Place Activated BatchNorm (InPlace-ABN) is a novel
Summary Explorer Summary Explorer is a tool to visually inspect the summaries from several state-of-the-art neural summarization models across multipl
segmentation TensorFlow implementation of ENet (https://arxiv.org/pdf/1606.02147.pdf) based on the official Torch implementation (https://github.com/e
Asynchronous Advantage Actor-Critic in PyTorch This is PyTorch implementation of A3C as described in Asynchronous Methods for Deep Reinforcement Learn
kalman-filter-in-single-object-tracking A simple implementation of Kalman filter in single object tracking https://www.bilibili.com/video/BV1Qf4y1J7D4
Learning to Teach with Dynamic Loss Functions This repo contains the simple demo for the NeurIPS-18 paper: Learning to Teach with Dynamic Loss Functio
TensorFlow2-GAN Collection of tf2.0 implementations of Generative Adversarial Network varieties presented in research papers. Model architectures will
Ἀνατομή is a PyTorch library to analyze representation of neural networks
MIA GNN Project Starter If you meet the version mismatch error for Lasagne library, please use following command to upgrade Lasagne library. pip insta
Neural-Pull: Learning Signed Distance Functions from Point Clouds by Learning to Pull Space onto Surfaces(ICML 2021) This repository contains the code
Feature Forge This library provides a set of tools that can be useful in many machine learning applications (classification, clustering, regression, e
GSL - Zero-shot Synthesis with Group-Supervised Learning Figure: Zero-shot synthesis performance of our method with different dataset (iLab-20M, RaFD,
SSD: A Unified Framework for Self-Supervised Outlier Detection [ICLR 2021] Pdf: https://openreview.net/forum?id=v5gjXpmR8J Code for our ICLR 2021 pape
Phase-in-NN This is the code for our project at Princeton (co-authors: Yuqi Nie, Hui Yuan). The paper title is: "Neural Network is heterogeneous: Phas
2 Jul 07, 2022
19 Dec 28, 2022
1.9k Jan 08, 2023
16 Dec 22, 2022
5 Jul 21, 2022
76 Jan 01, 2023
1.3k Dec 29, 2022
42 Aug 14, 2022
248 Dec 16, 2022
38 Dec 12, 2022
130 Dec 26, 2022
15 Dec 30, 2021
41 Apr 28, 2022
50 Dec 05, 2022
6 Nov 09, 2022
149 Dec 15, 2022
380 Nov 05, 2022
62 Dec 21, 2022
113 Nov 27, 2022
1 Nov 04, 2021