modeci_mdf.interfaces.pytorch.exporter.generate_main_forward(nodes: List[node], execution_order: List[str], d_e: Dict[str, Any])[source]

Helper function to generate the main forward method that will specify the execution of the pytorch model. This requires proper ordering of module calls as well as preservation of variables.

  • nodes – list of nodes in the graph

  • execution_order – List of nodes in the order of execution

  • d_e – Weights on edges stored in dict format

Returns: Function returns the main forward call that will be used to define pytorch model