提交 ec512cdc 编写于 作者: L Luo Tao

add comment for branch network

上级 f45818e7
...@@ -81,6 +81,9 @@ class InferenceTranspiler: ...@@ -81,6 +81,9 @@ class InferenceTranspiler:
current_op = self.block.ops[i] current_op = self.block.ops[i]
# TODO(luotao1): consider only conv2d now. fc would be delt later. # TODO(luotao1): consider only conv2d now. fc would be delt later.
if current_op.type in ['conv2d']: 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] next_op = self.block.ops[i + 1]
# conv2d without bias # conv2d without bias
if (next_op.type == 'batch_norm'): if (next_op.type == 'batch_norm'):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册