TylerYep/torchinfo
quality grade B, 78 out of 100View model summaries in PyTorch!
- stars
- 3.0k
- stars gained this week
- —this week
- forks, open issues and contributors
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Core deep-learning frameworks and libraries for pretraining and distributed training.
Signals: deep-learning, neural-network, pytorch, tensorflow, jax, distributed-training, training, deepspeed
2,672 results
View model summaries in PyTorch!
PyTorch implementation of TabNet paper : https://arxiv.org/pdf/1908.07442.pdf
Image to prompt with BLIP and CLIP
FFCV: Fast Forward Computer Vision (and other ML workloads!)
PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)
DeepLab v3+ model in PyTorch. Support different backbones.
MOT using deepsort and yolov3 with pytorch
Deep recommender models using PyTorch.
pytorch tutorial for beginners
Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)
Open Source Pre-training Model Framework in PyTorch & Pre-trained Model Zoo
The state-of-the-art image restoration model without nonlinear activation functions.
A OpenMMLAB toolbox for human pose estimation, skeleton-based action recognition, and action synthesis.
Pytorch implementation of the Graph Attention Network model by Veličković et. al (2017, https://arxiv.org/abs/1710.10903)
A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.
深度学习100例、深度学习DL、图片分类、目标识别、目标检测、自然语言处理nlp、文本分类、TensorFlow、PyTorch
torch-optimizer -- collection of optimizers for Pytorch
Siamese and triplet networks with online pair/triplet mining in PyTorch
Official implementation of CVPR2020 paper "VIBE: Video Inference for Human Body Pose and Shape Estimation"
ResNeSt: Split-Attention Networks
OpenMMLab Self-Supervised Learning Toolbox and Benchmark
Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2
Semantic Segmentation Architectures Implemented in PyTorch
:fire: Machine Learning Notebooks
24,540 repositories in the index in total.