diff --git a/model_zoo/official/recommend/wide_and_deep/src/wide_and_deep.py b/model_zoo/official/recommend/wide_and_deep/src/wide_and_deep.py
index 149908185fd..0dda5d58e02 100644
--- a/model_zoo/official/recommend/wide_and_deep/src/wide_and_deep.py
+++ b/model_zoo/official/recommend/wide_and_deep/src/wide_and_deep.py
@@ -212,7 +212,6 @@ class WideDeepModel(nn.Cell):
                 self.deep_embeddinglookup = nn.EmbeddingLookup(self.vocab_size, self.emb_dim, target=target,
                                                                slice_mode=nn.EmbeddingLookup.TABLE_COLUMN_SLICE)
                 self.dense_layer_1.dropout.dropout.shard(((1, get_group_size()),))
-                self.dense_layer_1.dropout.dropout_do_mask.shard(((1, get_group_size()),))
                 self.dense_layer_1.matmul.shard(((1, get_group_size()), (get_group_size(), 1)))
                 self.dense_layer_1.matmul.add_prim_attr("field_size", self.field_size)
                 self.deep_mul.shard(((1, 1, get_group_size()), (1, 1, 1)))
@@ -234,7 +233,6 @@ class WideDeepModel(nn.Cell):
             self.wide_mul.shard(((1, get_group_size(), 1), (1, get_group_size(), 1)))
             self.reduce_sum.shard(((1, get_group_size(), 1),))
             self.dense_layer_1.dropout.dropout.shard(((1, get_group_size()),))
-            self.dense_layer_1.dropout.dropout_do_mask.shard(((1, get_group_size()),))
             self.dense_layer_1.matmul.shard(((1, get_group_size()), (get_group_size(), 1)))
             self.embedding_table = self.deep_embeddinglookup.embedding_table
         elif parameter_server: