pytorch-utils

There are 6 repositories under pytorch-utils topic.

sovrasov/flops-counter.pytorch
Flops counter for convolutional networks in pytorch framework
Language:Python2.8k 15 98306
frgfm/torch-scan
Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
Language:Python213 7 1722
emaballarin/ebtorch
🤓🔥 Collection of PyTorch additions, extensions, utilities, uses and abuses
Language:Python9 4 01
emaballarin/pyromaniac
:fire: Collection of PyTorch uses and abuses
Language:Python1 2 1
STomoya/storch
pyTORCH utilities for STomoya
Language:Python0 2 200
ichko/ez-torch
Utility functions and fluent interface for PyTorch tensors and modules
Language:Python1 1