zxd1997066 Goto Github PK

accelerate

🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision

api

attention-module

Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"

benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

chemicalx

A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD 2022)

conformer

PyTorch implementation of "Conformer: Convolution-augmented Transformer for Speech Recognition" (INTERSPEECH 2020)

convnext

Code release for ConvNeXt model

crnn.pytorch

Convolutional recurrent network in pytorch

data

A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.

dcrnn

Diffusion Convolutional Recurrent Neural Network Implementation in PyTorch

deepspeech.pytorch

Speech Recognition using DeepSpeech2.

deit

Official DeiT repository

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

dlrm

An implementation of a deep learning recommendation model (DLRM)

fastspeech2

An implementation of Microsoft's "FastSpeech 2: Fast and High-Quality End-to-End Text to Speech"

hbonet

[ICCV 2019] Harmonious Bottleneck on Two Orthogonal Dimensions, surpassing MobileNetV2

hybridnets

HybridNets: End-to-End Perception Network

malconv-pytorch

Pytorch implementation of MalConv

metaformer

A PyTorch implementation of "MetaFormer: A Unified Meta Framework for Fine-Grained Recognition". A reference PyTorch implementation of “CoAtNet: Marrying Convolution and Attention for All Data Sizes”

mganet-dcc2020

PyTorch implementation of "MGANet & DCC2020 paper"

midas

Code for robust monocular depth estimation described in "Ranftl et. al., Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfer, TPAMI 2022"

mlperf_training

Reference implementations of MLPerf™ training benchmarks

oob-common

pretrained-models.pytorch

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch-3dunet

3D U-Net model for volumetric semantic segmentation written in pytorch

pytorch-examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.

pytorch-unet

Simple PyTorch implementations of U-Net/FullyConvNet (FCN) for image segmentation