oneflow.nn.graph.graph\_block.GraphModule.set\_stage ==================================================== .. currentmodule:: oneflow.nn.graph.graph_block .. automethod:: GraphModule.set_stage