Skip to content

polish parallel dygraph code#17164

Merged
chengduoZH merged 8 commits intoPaddlePaddle:developfrom
Yancey0623:parallel_dygraph_unittest
May 17, 2019
Merged

polish parallel dygraph code#17164
chengduoZH merged 8 commits intoPaddlePaddle:developfrom
Yancey0623:parallel_dygraph_unittest

Conversation

@Yancey0623
Copy link
Contributor

@Yancey0623 Yancey0623 commented Apr 29, 2019

  1. Polish Parallel Dygraph code.
  2. add se-resnext unit test.

@Yancey0623 Yancey0623 changed the title add var grad hook polish parallel dygraph code May 7, 2019
@Yancey0623 Yancey0623 requested a review from chengduoZH May 7, 2019 09:32

class DataParallel(layers.Layer):
def __init__(self, layers):
def __init__(self, layers, strategy):
Copy link
Contributor

Choose a reason for hiding this comment

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

Please add doc for DataParallel.

loss = loss / loss_scale
return loss

def apply_collective_grads(self):
Copy link
Contributor

Choose a reason for hiding this comment

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

Please add doc for the above functions, i.e. forward , scale_loss and apply_collective_grads .

@chengduoZH chengduoZH merged commit 0217555 into PaddlePaddle:develop May 17, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants