def __init__(
self,
all_checkpointers: List[BaseCheckpointer] = None,
):
BaseStep.__init__(self)
ResumableStepMixin.__init__(self)
IdentityHandlerMethodsMixin.__init__(self)
self.all_checkpointers = all_checkpointers
def __init__(self, wrapped: BaseTransformer, then_unflatten: bool = True):
MetaStep.__init__(self, wrapped)
ResumableStepMixin.__init__(self)
ForceHandleMixin.__init__(self)
self.then_unflatten = then_unflatten
self.len_di = []
self.len_eo = []
def __init__(self, wrapped: BaseStep):
BaseStep.__init__(self)
MetaStepMixin.__init__(self, wrapped)
ResumableStepMixin.__init__(self)
def __init__(self, force_checkpoint_name: str = None):
ResumableStepMixin.__init__(self)
BaseStep.__init__(self)
self.force_checkpoint_name = force_checkpoint_name
def __init__(self, wrapped: BaseTransformer):
MetaStep.__init__(self, wrapped)
ResumableStepMixin.__init__(self)