modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints.StreamingCSLeaf#

class modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints.StreamingCSLeaf[source]#

Bases: object

Marks checkpoint keys that can be directly loaded from/saved to the H5 checkpoint. Non-leafs are accessed through StreamingCSWriterView due to their iterable nature.

Methods

previous

modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints.dtype_byte_size

next

modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints.StreamingCSWriter