pytorch-extension

There are 13 repositories under pytorch-extension topic.

ildoonet/pytorch-gradual-warmup-lr
Gradually-Warmup Learning Rate Scheduler for PyTorch
Language:Python991 9 18126
stevewongv/SPANet
Spatial Attentive Single-Image Deraining with a High Quality Real Rain Dataset (CVPR'19)
Language:Jupyter Notebook226 11 2455
stevewongv/DSC-PyTorch
A PyTorch implementation of "Direction-Aware Spatial Context Features for Shadow Detection" CVPR'18 | T-PAMI'19
Language:Python94 2 1832
cmpark0126/pytorch-polynomial-lr-decay
Polynomial Learning Rate Decay Scheduler for PyTorch
Language:Python66 0 413
ZichaoLong/aTEAM
A pyTorch Extension for Applied Mathematics
Language:Python40 3 211
aredden/torch-bnb-fp4
Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops
Language:Python29 2 01
frgfm/torch-cuda-template
Template for CUDA / C++ extension writing with PyTorch
Language:Python6 1 00
openml/openml-pytorch
Pytorch extension for openml-python
Language:Jupyter Notebook6 5 160
artitw/apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
Language:Python5 2 10
pminhtam/xnor_conv_pytorch_extension
XNOR-Net with binary conv2d kernels with XNOR GEMM op, support both CPU and GPU.
Language:C5 1 00
djalex88/gaussian_rbf
PyTorch extension
Language:Cuda2 1 00
AmirMardan/pytorch_extending_cpp_binding
Binding C++ to PyTorch and extending PyTorch
Language:Python1 1 00
eshibusawa/Simple-Examples
simple examples of tools and libraries
Language:Python0 1 00