ksuit.initializers.base.initializer_base¶
Classes¶
Helper class that provides a standard way to create an ABC using |
Module Contents¶
- class ksuit.initializers.base.initializer_base.InitializerBase(path_provider)¶
Bases:
abc.ABCHelper class that provides a standard way to create an ABC using inheritance.
Base class for model initializers.
- Parameters:
path_provider (ksuit.providers.PathProvider) – PathProvider instance to access paths to load models from.
- logger¶
- path_provider¶
- abstractmethod init_weights(model)¶
Initialize the model weights from the checkpoint.
- Parameters:
model (ksuit.models.model_base.ModelBase) – the model to load the weights into.
- Return type:
None
- abstractmethod init_optimizer(model)¶
Initialize the optimizer for the model.
- Parameters:
model (ksuit.models.model_base.ModelBase) – a model to initialize the optimizer for. Assumes the model has an attribute optim.
- Return type:
None