fix pytorch-to-hls for sublayer pr

FPGA4HEP · Aug 10, 2018 · 4b5aca3 · 4b5aca3
1 parent 91f8db1
commit 4b5aca3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pytorch-to-hls/pytorch-to-hls.py b/pytorch-to-hls/pytorch-to-hls.py
@@ -82,6 +82,9 @@ def main():
         layer["n_in"] =  int(matchname.group(2))
         layer["n_out"] =  int(matchname.group(3))
 
+        # number of sublayer calls
+        layer["n_part"] = 1
+
         # #Extract type of activation and number of nodes
         layer["activation"] = modelstr[i+1].split(":")[-1].strip().lower()[:-2]