recover

x2paddle/core/program.py

@@ -285,14 +285,14 @@ 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:
-            self.gen_dygraph_code(save_dir)
-            self.dump_dygraph_parameter(save_dir)
+            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)
         # 动转静
         code_path = osp.join(osp.abspath(save_dir), "x2paddle_code.py")
         print("Exporting inference model from python code ('{}')... \n".format(code_path))