未验证 提交 7408ada6 编写于 作者: J Jason 提交者: GitHub

Merge pull request #741 from wjj19950828/Fixed_pytorch_readme

Fixed FusedBatchNorm op
......@@ -563,7 +563,6 @@ class TFOpMapper():
n, h, w, c = input.out_shapes[0]
else:
n, c, h, w = input.out_shapes[0]
self.params["{}_{}".format(node.name, gamma.name)] = self.params[
gamma.name]
self.params["{}_{}".format(node.name, beta.name)] = self.params[
......@@ -584,7 +583,8 @@ class TFOpMapper():
moving_mean.name)),
moving_variance_name=string("{}_{}".format(node.name,
moving_var.name)),
is_test=True)
is_test=True,
trainable_statistics=node.get_attr("is_training"))
if data_format == "NHWC":
self.paddle_graph.add_layer(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册