p04896132
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!4884 fix wide&deep dropout set strategy 

Merge pull request !4884 from yao_yf/fix_wide_and_deep_dropout
This commit is contained in:
mindspore-ci-bot 2020-08-21 11:14:53 +08:00 committed by Gitee
parent a64d00b4b9 1e60cef198
commit 7ba977c435
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/nn/layer/basic.py
View File

@ -93,9 +93,7 @@ class Dropout(Cell):
self.dropout_do_mask = P.DropoutDoMask()
self.cast = P.Cast()
self.is_gpu = context.get_context('device_target') in ["GPU"]
if self.is_gpu:
self.dropout = P.Dropout(keep_prob)
self.dropout = P.Dropout(keep_prob)
def construct(self, x):
if not self.training:

6
mindspore/nn/layer/embedding.py
View File

@ -128,8 +128,10 @@ class EmbeddingLookup(Cell):
vocab_size (int): Size of the dictionary of embeddings.
embedding_size (int): The size of each embedding vector.
param_init (str): The initialize way of embedding table. Default: 'normal'.
target (str): Specify the target where the op is executed. Default: 'CPU'.
slice_mode (str): The slicing way in semi auto parallel/auto parallel. Default: 'batch_slice'.
target (str): Specify the target where the op is executed. The value should in
['DEVICE', 'CPU']. Default: 'CPU'.
slice_mode (str): The slicing way in semi auto parallel/auto parallel. The value should get through
nn.EmbeddingLookUpSplitMode. Default: 'batch_slice'.
manual_shapes (tuple): The accompaniment array in field slice mode.
Inputs: