summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--run.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/run.py b/run.py
index dc61660..9c0bae2 100644
--- a/run.py
+++ b/run.py
@@ -57,8 +57,8 @@ def process_tree(moduleNetwork, a, b, tensorOutput, steps):
if steps < 2:
return [tensorMiddle]
else:
- tensorLeft = process_tree(moduleNetwork, tensorInputFirst, tensorMiddle, tensorOutput, steps / 2)
- tensorRight = process_tree(moduleNetwork, tensorMiddle, tensorInputSecond, tensorOutput, steps / 2)
+ tensorLeft = process_tree(moduleNetwork, a, tensorMiddle, tensorOutput, steps / 2)
+ tensorRight = process_tree(moduleNetwork, tensorMiddle, b, tensorOutput, steps / 2)
return tensorLeft + [tensorMiddle] + tensorRight
if arguments_strVideo and arguments_strVideoOut: