Awesome Open Source

Programming Languages

Search results for pytorch pretrained models

pretrained-models x

154 search results found

Transformers ⭐ 124,049

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Pytorch Image Models ⭐ 29,680

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Segmentation_models.pytorch ⭐ 8,226

Segmentation models with pretrained backbones. PyTorch.

Petals ⭐ 8,040

🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading

Lmflow ⭐ 7,589

An Extensible Toolkit for Finetuning and Inference of Large Foundation Models. Large Models for All.

Open_clip ⭐ 7,355

An open source implementation of CLIP.

Efficientnet Pytorch ⭐ 6,577

A PyTorch implementation of EfficientNet and EfficientNetV2 (coming soon!)

A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)

Pyannote Audio ⭐ 4,460

Neural building blocks for speaker diarization: speech activity detection, speaker change detection, overlapped speech detection, speaker embedding

Super Gradients ⭐ 4,125

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.

Silero Models ⭐ 4,088

Silero Models: pre-trained speech-to-text, text-to-speech and text-enhancement models made embarrassingly simple

Openprompt ⭐ 4,006

An Open-Source Framework for Prompt-Learning.

Superduperdb ⭐ 3,924

🔮 SuperDuperDB: Bring AI to your database! Build, deploy and manage any AI application directly with your existing data infrastructure, without moving your data. Including streaming inference, scalable model training and vector search.

Facenet Pytorch ⭐ 3,809

Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models

Efficient Ai Backbones ⭐ 3,770

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.

Albert_zh ⭐ 3,723

A LITE BERT FOR SELF-SUPERVISED LEARNING OF LANGUAGE REPRESENTATIONS, 海量中文预训练ALBERT模型

Fengshenbang Lm ⭐ 3,670

Fengshenbang-LM(封神榜大模型)是IDEA研究院认知计算与自然语言研究中心主导的大模型

中文语言理解测评基准 Chinese Language Understanding Evaluation Benchmark: datasets, baselines, pre-trained models, corpus and leaderboard

Pinto_model_zoo ⭐ 3,121

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.

Mmpretrain ⭐ 3,114

OpenMMLab Pre-training Toolbox and Benchmark

Chinese Clip ⭐ 2,816

Chinese version of CLIP which achieves Chinese cross-modal retrieval and representation generation.

Imgclsmob ⭐ 2,399

Sandbox for training deep learning networks

Funasr ⭐ 2,315

A Fundamental End-to-End Speech Recognition Toolkit and Open Source SOTA Pretrained Models.

Asteroid ⭐ 2,018

The PyTorch-based audio source separation toolkit for researchers

Easynlp ⭐ 1,871

EasyNLP: A Comprehensive and Easy-to-use NLP Toolkit

🏡 Fast & easy transfer learning for NLP. Harvesting language models for the industry. Focus on Question Answering.

Gen Efficientnet Pytorch ⭐ 1,561

Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS

Open3d Ml ⭐ 1,530

An extension of Open3D to address 3D Machine Learning tasks

Huggingface_hub ⭐ 1,462

The official Python client for the Huggingface Hub.

Cogview ⭐ 1,445

Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".

Spark ⭐ 1,355

[ICLR'23 Spotlight🔥] The first successful BERT/MAE-style pretraining on any convolutional network; Pytorch impl. of "Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling"

Chinese Electra ⭐ 1,253

Pre-trained Chinese ELECTRA（中文ELECTRA预训练模型）

Conv Emotion ⭐ 1,146

This repo contains implementation of different architectures for emotion recognition in conversations.

Involution ⭐ 1,106

[CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator

Cv Pretrained Model ⭐ 991

A collection of computer vision pre-trained models.

Textbox ⭐ 966

TextBox 2.0 is a text generation library with pre-trained language models

Bert Ner ⭐ 872

Pytorch-Named-Entity-Recognition-with-BERT

Large-scale pretrained models for goal-directed dialog

Breast_cancer_classifier ⭐ 804

Deep Neural Networks Improve Radiologists' Performance in Breast Cancer Screening

Swissarmytransformer ⭐ 803

SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.

Trt_pose ⭐ 763

Real-time pose estimation accelerated with NVIDIA TensorRT

Official SRFlow training code: Super-Resolution using Normalizing Flow in PyTorch

Pocket-Sized Multimodal AI for content understanding and generation across multilingual texts, images, and 🔜 video, up to 5x faster than OpenAI CLIP and LLaVA 🖼️ & 🖋️

Self Driving Car In Video Games ⭐ 729

A deep neural network that learns to drive in video games

Classification_models ⭐ 706

Classification models trained on ImageNet. Keras.

Pytorchinsight ⭐ 678

a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results

Cogview2 ⭐ 669

official code repo for paper "CogView2: Faster and Better Text-to-Image Generation via Hierarchical Transformers"

Patrickstar ⭐ 631

PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.

Diffwave ⭐ 628

DiffWave is a fast, high-quality neural vocoder and waveform synthesizer.

EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation

Ritm_interactive_segmentation ⭐ 592

Reviving Iterative Training with Mask Guidance for Interactive Segmentation

Pytorch Cnn Finetune ⭐ 580

Fine-tune pretrained Convolutional Neural Networks with PyTorch

Fbrs_interactive_segmentation ⭐ 579

[CVPR2020] f-BRS: Rethinking Backpropagating Refinement for Interactive Segmentation https://arxiv.org/abs/2001.10331

Functional Zoo ⭐ 557

PyTorch and Tensorflow functional model definitions

Treasure Of Transformers ⭐ 541

💁 Awesome Treasure of Transformers Models for Natural Language processing contains papers, videos, blogs, official repo along with colab Notebooks. 🛫☑️

Pytorch_cifar10 ⭐ 520

Pretrained TorchVision models on CIFAR10 dataset (with weights)

Pro_gan_pytorch ⭐ 504

Unofficial PyTorch implementation of the paper titled "Progressive growing of GANs for improved Quality, Stability, and Variation"

Wama_modules ⭐ 501

A PyTorch Computer Vision (CV) module library for building n-D networks flexibly ~

Neural Backed Decision Trees ⭐ 445

Making decision trees competitive with neural networks on CIFAR10, CIFAR100, TinyImagenet200, Imagenet

Fullsubnet ⭐ 443

PyTorch implementation of "FullSubNet: A Full-Band and Sub-Band Fusion Model for Real-Time Single-Channel Speech Enhancement."

Bertwithpretrained ⭐ 428

An implementation of the BERT model and its related downstream tasks based on the PyTorch framework

Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models

[CVPR 2020 Oral] PyTorch implementation of "Disentangling and Unifying Graph Convolutions for Skeleton-Based Action Recognition"

CRSLab is an open-source toolkit for building Conversational Recommender System (CRS).

Azureml Bert ⭐ 384

End-to-End recipes for pre-training and fine-tuning BERT using Azure Machine Learning Service

Mobilenetv2.pytorch ⭐ 352

72.8% MobileNetV2 1.0 model on ImageNet and a spectrum of pre-trained MobileNetV2 models

Dialogrpt ⭐ 325

EMNLP 2020: "Dialogue Response Ranking Training with Large-Scale Human Feedback Data"

Vision Aided Gan ⭐ 321

Ensembling Off-the-shelf Models for GAN Training (CVPR 2022 Oral)

Cartoonify ⭐ 313

Deploy and scale serverless machine learning app - in 4 steps.

Glasses ⭐ 311

High-quality Neural Networks for Computer Vision 😎

Tensorflow Model Zoo.torch ⭐ 270

InceptionV3, InceptionV4, Inception-Resnet pretrained models for Torch7 and PyTorch

Pytorch Human Pose Estimation ⭐ 268

Implementation of various human pose estimation models in pytorch on multiple datasets (MPII & COCO) along with pretrained models

PERT: Pre-training BERT with Permuted Language Model

Wavegrad ⭐ 239

A fast, high-quality neural vocoder.

Pytorch Nlu ⭐ 226

Pytorch-NLU，一个中文文本分类、序列标注工具包，支持中文长文本、短文本的多类、多标签分类任 Ptorch NLU, a Chinese text classification and sequence annotation toolkit, supports multi class and multi label classification tasks of Chinese long text and short text, and supports sequence annotation tasks such as Chinese named entity recognition, part of speech tagging and word segmentation.

Neural Nearest Neighbors Networks (NIPS*2018)

Bert Squad ⭐ 217

SQuAD Question Answering Using BERT, PyTorch

Searching prompt modules for parameter-efficient transfer learning.

Rust Mlops Template ⭐ 196

A work in progress to build out solutions in Rust for MLOPs

Frustum Convnet ⭐ 196

The PyTorch Implementation of F-ConvNet for 3D Object Detection

Mayavoz ⭐ 193

Pytorch based speech enhancement toolkit.

Code and pretrained models for LIP: Local Importance-based Pooling (ICCV 19)

Pytorch Cifar Models ⭐ 174

Pretrained models on CIFAR10/100 in PyTorch

Hugsvision ⭐ 170

HugsVision is a easy to use huggingface wrapper for state-of-the-art computer vision

Baby A3c ⭐ 153

A high-performance Atari A3C agent in 180 lines of PyTorch

Gan Vae Pretrained Pytorch ⭐ 148

Pretrained GANs + VAEs + classifiers for MNIST/CIFAR in pytorch.

Facenet Pytorch Glint360k ⭐ 140

A PyTorch implementation of the 'FaceNet' paper for training a facial recognition model with Triplet Loss using the glint360k dataset. A pre-trained model using Triplet Loss is available for download.

Lamda Pilot ⭐ 134

🎉 PILOT: A Pre-trained Model-Based Continual Learning Toolbox

Breast_density_classifier ⭐ 128

Breast density classification with deep convolutional neural networks

Ganimation_replicate ⭐ 126

An Out-of-the-Box Replicate of GANimation using PyTorch, pretrained weights are available!

Mff Pytorch ⭐ 124

Motion Fused Frames implementation in PyTorch, codes and pretrained models.

Tacotron2 Pytorch ⭐ 115

Yet another PyTorch implementation of Tacotron 2 with reduction factor and faster training speed.

中文语音识别系列，读者可以借助它快速训练属于自己的中文语音识别模型，或直接使用预训练模型测试效果。

Official PyTorch implementation of GDWCT (CVPR 2019)

Minirbt ⭐ 102

MiniRBT (中文小型预训练模型系列)

Const_layout ⭐ 99

Official implementation of the MM'21 paper "Constrained Graphic Layout Generation via Latent Optimization" (LayoutGAN++, CLG-LO, and Layout evaluation)

Ontoprotein ⭐ 98

Code and datasets for the ICLR2022 paper "OntoProtein: Protein Pretraining With Gene Ontology Embedding"

Imagenet ⭐ 97

Pytorch Imagenet Models Example + Transfer Learning (and fine-tuning)

Nfnets_pytorch ⭐ 94

Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".

Publaynet ⭐ 93

MaskRCNN on PubLayNet datasets. Paragraph detection, table detection, figure detection,...

Related Searches

Python Pytorch (15,943)

Deep Learning Pytorch (7,533)

Jupyter Notebook Pytorch (4,892)

Machine Learning Pytorch (2,934)

Dataset Pytorch (1,848)

Pytorch Convolutional Neural Networks (1,777)

Pytorch Neural Network (1,631)

Pytorch Natural Language Processing (1,408)

Network Pytorch (1,299)

Pytorch Computer Vision (1,230)

1-100 of 154 search results

Privacy | About | Terms | Follow Us On Twitter

Copyright 2018-2024 Awesome Open Source. All rights reserved.