Skip to content

Commit ec512cd

Browse files
committed
add comment for branch network
1 parent f45818e commit ec512cd

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

python/paddle/fluid/inference_transpiler.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,9 @@ def fuse_batch_norm(self, program, scope, place):
8181
current_op = self.block.ops[i]
8282
# TODO(luotao1): consider only conv2d now. fc would be delt later.
8383
if current_op.type in ['conv2d']:
84+
# TODO(luotao1): consider single chain network now.
85+
# For branch network, we counldn't use block.ops[i + 1] as
86+
# the judgment condition.
8487
next_op = self.block.ops[i + 1]
8588
# conv2d without bias
8689
if (next_op.type == 'batch_norm'):

0 commit comments

Comments
 (0)