Skip to content

In place batch_norm#9420

Merged
reyoung merged 2 commits intoPaddlePaddle:developfrom
tonyyang-svail:in_place_bn
Mar 28, 2018
Merged

In place batch_norm#9420
reyoung merged 2 commits intoPaddlePaddle:developfrom
tonyyang-svail:in_place_bn

Conversation

@tonyyang-svail
Copy link

fix #9416

  1. add in-place attr to bn
  2. make batch norm in-place in img_conv_group by default
@jacquesqiao
Copy link
Member

jacquesqiao commented Mar 28, 2018

Have you verified the correctness of inplace for batch_nrom? I looked at the code and think it has no problem.
This article says it verified the correctness https://blog.csdn.net/nus_cs/article/details/78126804.

Copy link
Member

@jacquesqiao jacquesqiao left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

Copy link
Collaborator

@reyoung reyoung left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is not the final solution for inplace. Just a hot fix for conv_bn_pool

@reyoung reyoung merged commit 47e4afb into PaddlePaddle:develop Mar 28, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

3 participants