mfai.pytorch.models

mfai.pytorch.models.load_from_settings_file(model_name, in_channels, out_channels, settings_path, input_shape)

Instanciate a model from a settings file with Schema validation.

Return type:

Module

Parameters:

• model_name (str)

• in_channels (int)

• out_channels (int)

• settings_path (Path)

• input_shape (tuple[int, ...])

• mfai.pytorch.models.gan_dgmr
• mfai.pytorch.models.llms
• mfai.pytorch.models.nlam