Functions for converting from MDF models to PyTorch


build_script(nodes, execution_order, ...)

Helper function to create and assemble text components necessary to specify importable model script.

func_args(exp, arg_dict)

generate_main_forward(nodes, ...)

Helper function to generate the main forward method that will specify the execution of the pytorch model.

get_module_declaration_text(name, node_dict, ...)

Helper function to generate string in

mdf_to_pytorch(mdf_model, model_input, ...)

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