Joeless / TorchEx Public

forked from Abyssaledge/TorchEx

Notifications You must be signed in to change notification settings
Fork 0
Star 1

Collection of PyTorch customized operators.

1 star 6 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 60 Commits
test		test
torchex		torchex
.gitignore		.gitignore
README.md		README.md
setup.py		setup.py

Repository files navigation

TorchEx

Collection of PyTorch customized operators for personal projects.

Supported GPU Operators:

Weighted NMS used in RangeDet.
IngroupIndices used in SST.
DynamicPointPooling (Group Correction) used in FSD.
Connected Component Labeling used in FSD.

Usage

With PyTorch installation, simply run pip install -e ..

About

Collection of PyTorch customized operators.

Report repository

Releases

No releases published

Packages

No packages published

Languages

Cuda 61.0%
C++ 21.6%
Python 17.4%