peal.architectures.predictors.load_model¶

peal.architectures.predictors.load_model(model_config, input_channels, output_channels, model_path, device)[source]¶

This function loads a model from a given path. :param model_config: The config of the model. :param input_channels: The number of input channels of the model. :param output_channels: The number of output channels of the model. :param model_path: The path to the model weights. :param device: The device the model is loaded on.

Returns:

The loaded model.

Parameters:

model_config (ArchitectureConfig)
input_channels (pydantic.PositiveInt)
output_channels (pydantic.PositiveInt)