Browse Source

parallel exec order bug fix

pull/14196/head
Margaret_wangrui 5 years ago
parent
commit
ce4b3e472d
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/nn/wrap/cell_wrapper.py

+ 1
- 0
mindspore/nn/wrap/cell_wrapper.py View File

@@ -352,6 +352,7 @@ class TrainOneStepCell(Cell):
weights = self.weights
loss = self.network(*inputs)
sens = P.Fill()(P.DType()(loss), P.Shape()(loss), self.sens)
sens = F.depend(sens, loss)
grads = self.grad(self.network, weights)(*inputs, sens)
grads = self.grad_reducer(grads)
loss = F.depend(loss, self.optimizer(grads))


Loading…
Cancel
Save