recover

x2paddle/core/program.py

@@ -285,12 +285,12 @@ class PaddleGraph(object):
             hierarchical_tree.save_source_files(save_dir)
             self.dump_dygraph_parameter(save_dir)
         else:
-#             if self.source_type == "pytorch":
-#                 from x2paddle.optimizer.pytorch_code_optimizer import ModuleGraph
-#                 module_graph = ModuleGraph(self)
-#                 module_graph.save_source_files(save_dir)
-#                 self.dump_dygraph_parameter(save_dir)
-#             else:
+            if self.source_type == "pytorch":
+                from x2paddle.optimizer.pytorch_code_optimizer import ModuleGraph
+                module_graph = ModuleGraph(self)
+                module_graph.save_source_files(save_dir)
+                self.dump_dygraph_parameter(save_dir)
+            else:
                 self.gen_dygraph_code(save_dir)
                 self.dump_dygraph_parameter(save_dir)
         # 动转静