cerebras.modelzoo.tools.checkpoint_converters.streaming_checkpoints.StreamingCSLeaf#

class cerebras.modelzoo.tools.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