cerebras.modelzoo.tools.checkpoint_converters.base_converter.EquivalentSubkey#

class cerebras.modelzoo.tools.checkpoint_converters.base_converter.EquivalentSubkey(a, b)[source]#

Bases: object

EquivalentSubkey defines the bidirectional relationship between subkeys of a model’s checkpoint. This class is simply a 2-tuple with index bounds checking.

For example if the normalization layer in one model is named “norm” and “ln” in the other, the relationship can be represented as EquivalentSubkey(“norm”, “ln”).

Methods