diff --git a/modules/util/util.py b/modules/util/util.py index 4aed787ef55c73fff3add136c180e164a3c204c9..d2eb362e8ebace3e6e0708e51fd61d6c9e3b7b42 100644 --- a/modules/util/util.py +++ b/modules/util/util.py @@ -452,8 +452,7 @@ def normalize(data, param, mean_std_dict, add_noise=False, noise_scale=0.01, see if add_noise: if seed is not None: np.random.seed(seed) - rnd = np.random.normal(loc=0, size=data.size) - rnd *= noise_scale + rnd = np.random.normal(loc=0, scale=noise_scale, size=data.size) data += rnd not_valid = np.isnan(data)