From 04a9c8574a6cf148190c983f06f7b3c910086a91 Mon Sep 17 00:00:00 2001 From: Giorgos Siolas Date: Fri, 13 Mar 2020 05:56:34 +0200 Subject: [PATCH] Update initializers.py --- efficientnet/initializers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/efficientnet/initializers.py b/efficientnet/initializers.py index 46cf7b3..00bf1bc 100644 --- a/efficientnet/initializers.py +++ b/efficientnet/initializers.py @@ -27,7 +27,7 @@ class EfficientConv2DKernelInitializer(Initializer): def __call__(self, shape, dtype=K.floatx(), **kwargs): kernel_height, kernel_width, _, out_filters = shape fan_out = int(kernel_height * kernel_width * out_filters) - return tf.random_normal( + return tf.random.normal( shape, mean=0.0, stddev=np.sqrt(2.0 / fan_out), dtype=dtype) @@ -63,7 +63,7 @@ def __call__(self, shape, dtype=K.floatx(), **kwargs): an initialization for the variable """ init_range = 1.0 / np.sqrt(shape[1]) - return tf.random_uniform(shape, -init_range, init_range, dtype=dtype) + return tf.random.uniform(shape, -init_range, init_range, dtype=dtype) conv_kernel_initializer = EfficientConv2DKernelInitializer()