subblock_stats

Modules

modelopt.torch.puzzletron.subblock_stats.calc_subblock_params_and_memory	Calculate memory usage and parameter counts for neural network subblocks.
modelopt.torch.puzzletron.subblock_stats.calc_subblock_stats	Calc subblock stats to compute memory and runtime statistics for subblocks.

Subblock statistics collection for Puzzletron.