cerebras.modelzoo.tools.checkpoint_converters.streaming_checkpoints.dtype_byte_size#

cerebras.modelzoo.tools.checkpoint_converters.streaming_checkpoints.dtype_byte_size(dtype: torch.dtype) → float[source]#

Returns the size (in bytes) occupied by one parameter of type dtype.

Example:

`py >>> dtype_byte_size(torch.float32) 4.0 `

previous

cerebras.modelzoo.tools.checkpoint_converters.streaming_checkpoints.convert_file_size_to_int

next

cerebras.modelzoo.tools.checkpoint_converters.streaming_checkpoints.OnDemandDictionaryConverter