sakura.model_controllers.extractor_controller.ExtractorController.load_checkpoint

ExtractorController.load_checkpoint(state_dict: dict)

Restore training state of the extractor from given checkpoint.

This function loads the extractor’s epochs, loss weights, model state, and optimizer state from the provided state dictionary.

Parameters:

state_dict (dict) – A compatible state dictionary containing the saved state of the extractor, expected from save_checkpoint()

Returns:

None