From 3f5f5ed361d151a5efa199072e8e887d02e99738 Mon Sep 17 00:00:00 2001 From: ceci3 <592712189@qq.com> Date: Tue, 19 Mar 2019 04:28:38 +0000 Subject: [PATCH] fix dropout doc --- python/paddle/fluid/layers/nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index b34b0848e..7a9500a2d 100644 --- a/python/paddle/fluid/layers/nn.py +++ b/python/paddle/fluid/layers/nn.py @@ -1316,7 +1316,7 @@ def dropout(x, 1. downgrade_in_infer(default), downgrade the outcome at inference - train: out = input * mask - - inference: out = input * (1 - dropout_prob) + - inference: out = input * (1.0 - dropout_prob) (mask is a tensor same shape with input, value is 0 or 1 ratio of 0 is dropout_prob) -- GitLab