sakura.utils.size_estimator

PyTorch Model Memory Estimator

This module provides tools for estimating the memory footprint of neural network models, including both parameter storage and activation memory during forward/backward passes.

Classes