sakura.utils.size_estimator.SizeEstimator.get_parameter_sizes

SizeEstimator.get_parameter_sizes()

Collect dimensions of all parameters in the model.

Returns:

None