fix: added handling for unknown layers

MLRichter · Jan 23, 2022 · 03f4c3a · 03f4c3a
1 parent 8d81d41
commit 03f4c3a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/rfa_toolbox/encodings/pytorch/ingest_architecture.py b/rfa_toolbox/encodings/pytorch/ingest_architecture.py
@@ -31,7 +31,10 @@ def make_graph(
     def find_name(i, self_input, suffix=None):
         if i == self_input:
             return suffix
-        cur = i.node().s("name")
+        try:
+            cur = i.node().s("name")
+        except RuntimeError:
+            return suffix + "-unknownType"
         if suffix is not None:
             cur = cur + "." + suffix
             # print(cur)