[fp16] fix bug when use drop block in fp16 (#4178)

b19807f8 · shangliang Xu · GitHub · df522565 · b19807f8
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppdet/modeling/layers.py ppdet/modeling/layers.py +1 -1

未找到文件。
--- a/ppdet/modeling/layers.py
+++ b/ppdet/modeling/layers.py
@@ -279,7 +279,7 @@ class DropBlock(nn.Layer):
            for s in shape:
                gamma *= s / (s - self.block_size + 1)

-            matrix = paddle.cast(paddle.rand(x.shape, x.dtype) < gamma, x.dtype)
+            matrix = paddle.cast(paddle.rand(x.shape) < gamma, x.dtype)
            mask_inv = F.max_pool2d(
                matrix,
                self.block_size,