From ec512cdcce5cf6e5952c7184db222f3d293d218d Mon Sep 17 00:00:00 2001 From: Luo Tao Date: Fri, 13 Apr 2018 18:45:09 +0800 Subject: [PATCH] add comment for branch network --- python/paddle/fluid/inference_transpiler.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/paddle/fluid/inference_transpiler.py b/python/paddle/fluid/inference_transpiler.py index 7b7bd899ea9..be8a627953c 100644 --- a/python/paddle/fluid/inference_transpiler.py +++ b/python/paddle/fluid/inference_transpiler.py @@ -81,6 +81,9 @@ class InferenceTranspiler: current_op = self.block.ops[i] # TODO(luotao1): consider only conv2d now. fc would be delt later. if current_op.type in ['conv2d']: + # TODO(luotao1): consider single chain network now. + # For branch network, we counldn't use block.ops[i + 1] as + # the judgment condition. next_op = self.block.ops[i + 1] # conv2d without bias if (next_op.type == 'batch_norm'): -- GitLab