From 16bf7fc7f106969e95a43ed286df60447b4204f8 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Fri, 15 May 2020 11:51:25 +0800 Subject: [PATCH] support for paddle1.8 --- ppcls/modeling/architectures/efficientnet.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ppcls/modeling/architectures/efficientnet.py b/ppcls/modeling/architectures/efficientnet.py index 082a1422..0a0ec2ea 100644 --- 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 -- GitLab