Awesome Open Source

Programming Languages

Search results for pytorch benchmark

152 search results found

Mmdetection ⭐ 26,886

OpenMMLab Detection Toolbox and Benchmark

Tianshou ⭐ 7,125

An elegant PyTorch deep reinforcement learning library.

Mmsegmentation ⭐ 6,974

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

Imagededup ⭐ 4,789

😎 Finding duplicate images made easy!

Mmpose ⭐ 4,587

OpenMMLab Pose Estimation Toolbox and Benchmark.

Silero Models ⭐ 4,088

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

Mmaction2 ⭐ 3,647

OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark

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

Sequence modeling benchmarks and temporal convolutional networks

Benchmarking Gnns ⭐ 2,196

Repository for benchmarking graph neural networks

Garage ⭐ 1,677

A toolkit for reproducible reinforcement learning research.

Avalanche ⭐ 1,583

Avalanche: an End-to-End Library for Continual Learning based on PyTorch.

A Heterogeneous Benchmark for Information Retrieval. Easy to use, evaluate your models across 15+ diverse IR datasets.

Slm Lab ⭐ 1,052

Modular Deep Reinforcement Learning framework in PyTorch. Companion library of the book "Foundations of Deep Reinforcement Learning".

Inference ⭐ 1,003

Reference implementations of MLPerf™ inference benchmarks

Medmnist ⭐ 903

[pip install medmnist] 18x Standardized Datasets for 2D and 3D Biomedical Image Classification

Torchmoji ⭐ 882

😇A pyTorch implementation of the DeepMoji model: state-of-the-art deep learning model for analyzing sentiment, emotion, sarcasm etc

AoE (AI on Edge，终端智能，边缘计算) 是一个终端侧AI集成运行时环境 (IRE)，帮助开发者提升效率。

Omnisafe ⭐ 831

OmniSafe is an infrastructural framework for accelerating SafeRL research.

Benchmark ⭐ 737

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

Overcooked_ai ⭐ 593

A benchmark environment for fully cooperative human-AI performance.

Textclassificationbenchmark ⭐ 577

A Benchmark of Text Classification in PyTorch

Openmixup ⭐ 538

CAIRI Supervised, Semi- and Self-Supervised Visual Representation Learning Toolbox and Benchmark

Pdebench ⭐ 522

PDEBench: An Extensive Benchmark for Scientific Machine Learning

Torchgpipe ⭐ 479

A GPipe implementation in PyTorch

Convert_torch_to_pytorch ⭐ 463

Convert torch t7 model to pytorch model and source.

Openstl ⭐ 450

OpenSTL: A Comprehensive Benchmark of Spatio-Temporal Predictive Learning

Powerful Benchmarker ⭐ 426

A library for ML benchmarking. It's powerful.

Pytorch Correlation Extension ⭐ 385

Custom implementation of Corrleation Module

DrQ: Data regularized Q

Ptranking ⭐ 353

Learning to Rank in PyTorch

Tasks Assessing Protein Embeddings (TAPE), a set of five biologically relevant semi-supervised learning tasks spread across different domains of protein biology.

Fedscale ⭐ 347

FedScale is a scalable and extensible open-source federated learning (FL) platform.

Benchmarking Keras Pytorch ⭐ 342

🔥 Reproducibly benchmarking Keras and PyTorch models

Opencv_transforms ⭐ 319

OpenCV implementation of Torchvision's image augmentations

Model Vs Human ⭐ 301

Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)

Fast PyTorch based DSP for audio and 1D signals

Pyhpc Benchmarks ⭐ 290

A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python 🚀

Scimlbenchmarks.jl ⭐ 279

Scientific machine learning (SciML) benchmarks, AI for science, and (differential) equation solvers. Covers Julia, Python (PyTorch, Jax), MATLAB, R

Lanczosnetwork ⭐ 271

Lanczos Network, Graph Neural Networks, Deep Graph Convolutional Networks, Deep Learning on Graph Structured Data, QM8 Quantum Chemistry Benchmark, ICLR 2019

Implementation of Principal Neighbourhood Aggregation for Graph Neural Networks in PyTorch, DGL and PyTorch Geometric

Naszilla ⭐ 269

Naszilla is a Python library for neural architecture search (NAS)

Openvqa ⭐ 225

A lightweight, scalable, and general framework for visual question answering research

CARLA: A Python Library to Benchmark Algorithmic Recourse and Counterfactual Explanation Algorithms

Volksdep ⭐ 214

volksdep is an open-source toolbox for deploying and accelerating PyTorch, ONNX and TensorFlow models with TensorRT.

Pytorchvideocompression ⭐ 211

PyTorch implementation and benchmark of Video Compression

Official PyTorch implementation of HANet (CVPR 2020)

Modelnet40 C ⭐ 184

Repo for "Benchmarking Robustness of 3D Point Cloud Recognition against Common Corruptions" https://arxiv.org/abs/2201.12296

Pytorch Retraining ⭐ 169

Transfer Learning Shootout for PyTorch's model zoo (torchvision)

Torch Scan ⭐ 164

Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)

Map Free Reloc ⭐ 156

[ECCV 2022] Map-free Visual Relocalization: Metric Pose Relative to a Single Image

Models Comparison.pytorch ⭐ 141

Code for the paper Benchmark Analysis of Representative Deep Neural Network Architectures

Optimum Benchmark ⭐ 140

A unified multi-backend utility for benchmarking Transformers and Diffusers with support for Optimum's arsenal of hardware optimizations/quantization schemes.

Vlmevalkit ⭐ 137

Open-source evaluation toolkit of large vision-language models (LVLMs), support GPT-4v, Gemini, QwenVLPlus, 30+ HF models, 15+ benchmarks

Small Object Detection Benchmark ⭐ 128

icip2022 paper: sahi benchmark on visdrone and xview datasets using fcos, vfnet and tood detectors

Changedetection ⭐ 125

A framework for change detection using PyTorch

Pytorch Gpu Benchmark ⭐ 119

Using the famous cnn model in Pytorch, we run benchmarks on various gpu.

Deep Visual Geo Localization Benchmark ⭐ 117

Official code for CVPR 2022 (Oral) paper "Deep Visual Geo-localization Benchmark"

Benchmarl ⭐ 115

A collection of MARL benchmarks based on TorchRL

Dr_loss ⭐ 111

PyTorch Implementation for DR Loss

Torchbench ⭐ 111

Easily benchmark machine learning models in PyTorch

Lstms.pth ⭐ 110

PyTorch implementations of LSTM Variants (Dropout + Layer Norm)

Co Learning Learning From Noisy Labels With Self Supervision ⭐ 107

The official implementation of the ACM MM'2021 paper Co-learning: Learning from noisy labels with self-supervision.

Proteinworkshop ⭐ 93

Benchmarking framework for protein representation learning. Includes a large number of pre-training and downstream task datasets, models and training/task utilities.

[RA-L2022] V2X-Sim Dataset and Benchmark

PArametrized Recommendation and Ai Model benchmark is a repository for development of numerous uBenchmarks as well as end to end nets for evaluation of training and inference platforms.

Simple A2c Ppo ⭐ 84

Actor-critic trained w PPO on OpenAI's Procgen Benchmark (PyTorch). Built from scratch.

Mujoco Benchmark ⭐ 78

Provide full reinforcement learning benchmark on mujoco environments, including ddpg, sac, td3, pg, a2c, ppo, library

SIXray : A Large-scale Security Inspection X-ray Benchmark in CVPR 2019

Rnn_benchmarks ⭐ 73

RNN benchmarks of pytorch, tensorflow and theano

Dm_nevis ⭐ 71

NEVIS'22: Benchmarking the next generation of never-ending learners

Mvp_benchmark ⭐ 68

MVP Benchmark for Multi-View Partial Point Cloud Completion and Registration

Maskrcnn_benchmark.cpp ⭐ 63

Implementation maskrcnn-benchmark, pytorch c++ frontend

Form2fit ⭐ 57

Train generalizable policies for kit assembly with self-supervised dense correspondence learning.

Softpool ⭐ 56

SoftPoolNet: Shape Descriptor for Point Cloud Completion and Classification, accepted in ECCV 2020 as oral

Convolutionalneuraloperator ⭐ 52

This repository is the official implementation of the paper Convolutional Neural Operators for robust and accurate learning of PDEs

Face Attribute Prediction ⭐ 52

Face Attribute Prediction on CelebA benchmark with PyTorch Implementation

Bert_ocr.pytorch ⭐ 51

Unofficial PyTorch implementation of 2D Attentional Irregular Scene Text Recognizer

Pytorch Benchmark ⭐ 47

Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption

Policy Adaptation During Deployment ⭐ 46

Training code and evaluation benchmarks for the "Self-Supervised Policy Adaptation during Deployment" paper.

Code and models for "Pano3D: A Holistic Benchmark and a Solid Baseline for 360 Depth Estimation", OmniCV Workshop @ CVPR21.

Dlinfbench ⭐ 42

CNN model inference benchmarks for some popular deep learning frameworks

Empower the quality enhancement approaches for compressed images/videos.

Sugar Crepe ⭐ 40

[NeurIPS 2023] A faithful benchmark for vision-language compositionality

Spatiotemporal_datasets ⭐ 39

Spatiotemporal datasets collected for network science, deep learning and general machine learning research.

Segmentation Networks Benchmark ⭐ 34

Evaluation framework for testing segmentation networks in Keras

Turnkeyml ⭐ 34

The AI insights toolchain

Kaggle Open Images 2019 Instance Segmentation ⭐ 32

Simple solution for Open Images 2019 - Instance Segmentation competition using maskrcnn-benchmark.

A benchmark for Salient Object Detection (SOD).

Revisiting Plms ⭐ 31

Exploring Evolution-aware & free protein language models as protein function predictors

Re Ranking For Vpr ⭐ 31

Code for "Are Local Features All You Need for Cross-Domain Visual Place Recognition?" CVPR IMW 2023.

Catalyst Rl Framework ⭐ 29

Catalyst.RL: A Distributed Framework for Reproducible RL Research

Mac Ml Speed Test ⭐ 27

A few quick scripts focused on testing TensorFlow/PyTorch/Llama 2 on macOS.

Repository for Deep Active Localization research and benchmarks

A Comprehensive and Versatile Open-Source Federated Learning Framework

Implementation of Directional Graph Networks in PyTorch and DGL

Ghost Deblurgan ⭐ 25

This is a lightweight GAN developed for real-time deblurring. The model has a super tiny size and a rapid inference time. The motivation is to boost marker detection in robotic applications, however, you may use it for other applications definitely.

ROCm Machine Learning and HPC Stack installer

Pytorch_compiler_tutorial ⭐ 23

Codebase associated with the PyTorch compiler tutorial

Cssegmentation ⭐ 23

CSSegmentation: An Open Source Continual Semantic Segmentation Toolbox Based on PyTorch.

Related Searches

Python Pytorch (15,943)

Deep Learning Pytorch (7,533)

Jupyter Notebook Pytorch (4,892)

Machine Learning Pytorch (2,934)

Python Benchmark (2,141)

Dataset Pytorch (1,848)

Pytorch Convolutional Neural Networks (1,777)

Pytorch Neural Network (1,631)

Pytorch Natural Language Processing (1,408)

Pytorch Computer Vision (1,230)

1-100 of 152 search results

Privacy | About | Terms | Follow Us On Twitter

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