compile_models

lib.infer.plugin_utils.compile_models(plugin: ExtractPlugin, modules: list[torch.nn.Module]) → None

Compile any Torch modules in the plugin’s model attribute

Parameters:

• plugin (ExtractPlugin) – The plugin containing Torch modules to be compiled

• modules (list[torch.nn.Module]) – The list of Torch modules contained within the plugin’s model attribute

Return type:

None