modeci_mdf.interfaces.pytorch.exporter.mdf_to_pytorch

modeci_mdf.interfaces.pytorch.exporter.mdf_to_pytorch(mdf_model: Model, model_input: str, eval_models: bool, version: str)

Function loads and returns a pytorch model for all models specified in an mdf file.

Parameters:

• mdf_model – model in MDF format

• eval_models – Set Evaluation of model to True or False

• version – MDF version

• model_input – input file name

Returns: Returns a dictionary where key = model name, value = pytorch model object