From da13f5980c908573cb3b2c542ad0e4884e5e92b6 Mon Sep 17 00:00:00 2001 From: chujinjin Date: Wed, 20 May 2020 14:51:32 +0800 Subject: [PATCH] fix dropout in pynative --- mindspore/nn/layer/basic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/layer/basic.py b/mindspore/nn/layer/basic.py index 115a0ec43c69..9b12d17c07b3 100644 --- a/mindspore/nn/layer/basic.py +++ b/mindspore/nn/layer/basic.py @@ -78,7 +78,7 @@ class Dropout(Cell): if keep_prob <= 0 or keep_prob > 1: raise ValueError("dropout probability should be a number in range (0, 1], but got {}".format(keep_prob)) validator.check_subclass("dtype", dtype, mstype.number_type, self.cls_name) - self.keep_prob = Tensor(keep_prob) + self.keep_prob = keep_prob self.seed0 = seed0 self.seed1 = seed1 self.dtype = dtype -- Gitee