yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!9777 fix auto parallet full batch 

From: @limingqi107
Reviewed-by: @chujinjin,@cristoval
Signed-off-by: @cristoval
This commit is contained in:
mindspore-ci-bot 2020-12-10 17:52:37 +08:00 committed by Gitee
parent 8c0d066bf9 42b01afc19
commit f7c339fce1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
model_zoo/official/recommend/wide_and_deep/src/wide_and_deep.py
View File

@ -438,8 +438,9 @@ class PredictWithSigmoid(nn.Cell):
self.network = network
self.sigmoid = P.Sigmoid()
parallel_mode = context.get_auto_parallel_context("parallel_mode")
full_batch = context.get_auto_parallel_context("full_batch")
is_auto_parallel = parallel_mode in (ParallelMode.SEMI_AUTO_PARALLEL, ParallelMode.AUTO_PARALLEL)
if is_auto_parallel:
if is_auto_parallel and full_batch:
self.sigmoid.shard(((1, 1),))
def construct(self, batch_ids, batch_wts, labels):