support for paddle1.8

2020-05-15 11:51:25 +08:00 · 2020-05-15 11:51:25 +08:00 · 16bf7fc7f1
parent a55f826683
commit 16bf7fc7f1
1 changed files with 3 additions and 2 deletions
--- a/ppcls/modeling/architectures/efficientnet.py
+++ b/ppcls/modeling/architectures/efficientnet.py
@ -192,8 +192,9 @@ class EfficientNet():
        if is_test:
            return inputs
        keep_prob = 1.0 - prob
-        random_tensor = keep_prob + fluid.layers.uniform_random_batch_size_like(
-            inputs, [-1, 1, 1, 1], min=0., max=1.)
+        inputs_shape = fluid.layers.shape(inputs)
+        random_tensor = keep_prob + fluid.layers.uniform_random(
+            shape=[inputs_shape[0], 1, 1, 1], min=0., max=1.)
        binary_tensor = fluid.layers.floor(random_tensor)
        output = inputs / keep_prob * binary_tensor
        return output