cerebras.modelzoo.tools.checkpoint_converters.base_converter.FormatIndices#

class cerebras.modelzoo.tools.checkpoint_converters.base_converter.FormatIndices[source]#

Bases: object

Class for propagating src and tgt version to base converter.

Methods

Attributes

direction

src_index

tgt_index

__init__(direction: int, src_index: int, tgt_index: int) None#