pytorch-utils

There are 5 repositories under pytorch-utils topic.

sovrasov/flops-counter.pytorch
Flops counter for neural networks in pytorch framework
Language:Python3k 15 101309
frgfm/torch-scan
Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
Language:Python222 7 1721
emaballarin/ebtorch
🤓🔥 Collection of PyTorch additions, extensions, utilities, uses and abuses
Language:Python9 1 11
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