pytorch-extension topic
List
pytorch-extension repositories
pytorch-gradual-warmup-lr
966
Stars
126
Forks
Watchers
Gradually-Warmup Learning Rate Scheduler for PyTorch
SPANet
199
Stars
53
Forks
Watchers
Spatial Attentive Single-Image Deraining with a High Quality Real Rain Dataset (CVPR'19)
DSC-PyTorch
90
Stars
30
Forks
Watchers
A PyTorch implementation of "Direction-Aware Spatial Context Features for Shadow Detection" CVPR'18 | T-PAMI'19
pytorch-polynomial-lr-decay
65
Stars
13
Forks
Watchers
Polynomial Learning Rate Decay Scheduler for PyTorch
aTEAM
36
Stars
9
Forks
Watchers
A pyTorch Extension for Applied Mathematics
torch-bnb-fp4
28
Stars
1
Forks
Watchers
Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops