Skip to content

loadweightsmodelfactory

loadweightsmodelfactory

Module for LoadWeightsModelFactory

Classes

LoadWeightsModelFactory 

LoadWeightsModelFactory(
    model_factory, weights_config, loading_options=None
)

Bases: ModelFactory

model factory to load weights before training starts

Source code in niceml/dlframeworks/keras/models/loadweightsmodelfactory.py 
16
17
18
19
20
21
22
23
24
def __init__(
    self,
    model_factory: ModelFactory,
    weights_config: LocationConfig,
    loading_options: Optional[dict] = None,
):
    self.model_factory = model_factory
    self.weights_config = weights_config
    self.loading_options = loading_options or {}

Functions