Fix bias import issue in import_tensorflow

CEA-LIST · May 26, 2020 · a3c572b · a3c572b
1 parent 9b0e7d2
commit a3c572b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/import_tensorflow.py b/tools/import_tensorflow.py
@@ -49,6 +49,8 @@ def exportFreeParameters(modelName, targetDir):
                 name[offset] = "scales"
             elif name[offset] == "beta":
                 name[offset] = "biases"
+            elif name[offset] == "bias":
+                name[offset] = "biases"
             elif name[offset] == "moving_mean":
                 name[offset] = "means"
             elif name[offset] == "moving_variance":