diff --git a/declearn/optimizer/modules/_noise.py b/declearn/optimizer/modules/_noise.py index f593a6bce116a0ef60bb51fc57c86bfb5d21660c..6aaf3ad96b2a696c3defc771bd00c054bc3f5bd7 100644 --- a/declearn/optimizer/modules/_noise.py +++ b/declearn/optimizer/modules/_noise.py @@ -95,7 +95,7 @@ class NoiseModule(OptiModule, metaclass=ABCMeta, register=False): for key in gradients.coefs } # Add the sampled noise to the gradients and return them. - # Silence warnings about sparse gradients getting sparsified. + # Silence warnings about sparse gradients getting densified. with warnings.catch_warnings(): warnings.filterwarnings("ignore", ".*densifying.*", RuntimeWarning) return gradients + NumpyVector(noise)