diff --git a/oneflow/core/functional/impl/nn_functor.cpp b/oneflow/core/functional/impl/nn_functor.cpp index 882c9c7d0fe..482a2ecb67c 100644 --- a/oneflow/core/functional/impl/nn_functor.cpp +++ b/oneflow/core/functional/impl/nn_functor.cpp @@ -1878,7 +1878,7 @@ class FusedBiasAddDropoutFunctor { axis_val += num_axes; } JUST(fused_bias_add_mask_attrs.SetAttr("axis", axis_val)); - if (p >= 0.0) { + if (p > 0.0) { return SequenceFunction()>([&]() -> Maybe { return OpInterpUtil::Dispatch( *random_mask_like_op_, {a},