Skip the error model for graph variable rename

Author: TelGome

graph_net/torch/graph_variable_renamer.py

@@ -75,19 +75,25 @@ def _make_config(
         }
 
     def __call__(self, rel_model_path):
-        src_model_path = os.path.join(self.config["model_path_prefix"], rel_model_path)
-        module, inputs = get_torch_module_and_inputs(src_model_path)
-        gm = parse_sole_graph_module(module, inputs)
-        gm = self.rename_graph_variables(gm, inputs, src_model_path)
-        dst_model_path = os.path.realpath(
-            os.path.join(self.config["output_dir"], rel_model_path)
-        )
-        Path(dst_model_path).parent.mkdir(parents=True, exist_ok=True)
-        shutil.copytree(src_model_path, dst_model_path, dirs_exist_ok=True)
-        self._update_model_py_file(gm, dst_model_path)
-        self._update_weight_meta_py_file(src_model_path, dst_model_path)
-        self._update_input_meta_py_file(src_model_path, dst_model_path)
-        self._try_run(dst_model_path)
+        try:
+            src_model_path = os.path.join(
+                self.config["model_path_prefix"], rel_model_path
+            )
+            module, inputs = get_torch_module_and_inputs(src_model_path)
+            gm = parse_sole_graph_module(module, inputs)
+            gm = self.rename_graph_variables(gm, inputs, src_model_path)
+            dst_model_path = os.path.realpath(
+                os.path.join(self.config["output_dir"], rel_model_path)
+            )
+            Path(dst_model_path).parent.mkdir(parents=True, exist_ok=True)
+            shutil.copytree(src_model_path, dst_model_path, dirs_exist_ok=True)
+            self._update_model_py_file(gm, dst_model_path)
+            self._update_weight_meta_py_file(src_model_path, dst_model_path)
+            self._update_input_meta_py_file(src_model_path, dst_model_path)
+            self._try_run(dst_model_path)
+        except Exception:
+            print("Failed to rename variables of ", src_model_path)
+            print("Skipping this model and continuing...\n")
 
     def _try_run(self, model_path):
         assert self.model_runnable_predicator(