Browse Source

!10045 [AutoParallel]Fix dropout log bug

From: @lichen666
Reviewed-by: @limingqi107,@stsuteng
Signed-off-by: @limingqi107,@stsuteng
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
665c58bc22
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/frontend/parallel/ops_info/activation_info.cc

+ 1
- 0
mindspore/ccsrc/frontend/parallel/ops_info/activation_info.cc View File

@@ -97,6 +97,7 @@ Status Activation::GenerateStrategies(int64_t stage_id) {
return FAILED;
}

is_auto_parallel_ = true;
Shape input0_split(inputs_shape_[0].size(), 1);
Shapes splittable_inputs = {input0_split};



Loading…
Cancel
Save