ONNX-TensorRT: TensorRT backend for ONNX

ONNX-TensorRT: TensorRT backend for ONNX

Related Repos

rajabzz digikala-sentiment-lstm 🧠 Trains a simple LSTM model on the Digikala product comment dataset for the sentiment classification task Installation Install python and pip. Create a virtualenv and activate

cornellius-gp GPyTorch News: GPyTorch v1.1 GPyTorch v1.1 has just been released. GPyTorch is a Gaussian process library implemented using PyTorch. GPyTorch is designed for creating scalable, flexible, and modular Gaussian process mode

rlworkgroup garage garage is a toolkit for developing and evaluating reinforcement learning algorithms, and an accompanying library of state-of-the-art implementations built using that toolkit. The toolkit provides wide range of modul

SauceCat PDPbox python partial dependence plot toolbox Motivation This repository is inspired by ICEbox. The goal is to visualize the impact of certain features towards model prediction for any supervised learning algorithm

mcleavey Clara: A Neural Net Music Generator Take the AI vs Human Quiz. Train an AWD-LSTM to generate piano or violin/piano music Project overview is here. Detailed paper is here. Requirements: PyTorch version 0.3.0.post4

kornia Kornia is a differentiable computer vision library for PyTorch. It consists of a set of routines and differentiable modules to solve generic computer vision problems. At its core, the package uses PyTorch as its main back

kindredresearch SenseAct: A computational framework for real-world robot learning tasks This repository provides the implementation of several reinforcement learning tasks with multiple real-world robots. These tasks come with an interface sim

BeloborodovDS Mobilenet+SSD face detector training This repo contains code for Mobilenet+SSD face detector training. This detector is compatible with Movidius Neural Compute Stick. You need NCSDK to test it with Neural Compute Stick.