modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter#
Classes
Converts between checkpoint state_dict formats. |
|
HF checkpoints contain model only while CS checkpoints package model, optimizer, and lr_scheduler into a single checkpoint. |
|
CS packages model, optimizer, and lr_scheduler into a single config. |
|
CS packages model, optimizer, and lr_scheduler into a single config. |
|
A dictionary converter represents a pair of two dictionary formats that can be converted between each other. |
|
ConversionRule defines a "rule" which: |
|
EquivalentSubkey defines the bidirectional relationship between subkeys of a model's checkpoint. |
|
Exceptions
Raised when a config cannot be converted |