From 805116250dbba3d507bcb97c9db3e52a68f9c7cc Mon Sep 17 00:00:00 2001 From: Aron Jansen Date: Thu, 25 Jan 2024 09:02:51 +0100 Subject: [PATCH] Rewording docstring Co-authored-by: Tanjona Rabemananjara --- n3fit/src/n3fit/backends/keras_backend/MetaModel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/n3fit/src/n3fit/backends/keras_backend/MetaModel.py b/n3fit/src/n3fit/backends/keras_backend/MetaModel.py index 5a78bf32e3..fde7c4f987 100644 --- a/n3fit/src/n3fit/backends/keras_backend/MetaModel.py +++ b/n3fit/src/n3fit/backends/keras_backend/MetaModel.py @@ -438,7 +438,7 @@ def is_stacked_single_replicas(layer): def get_layer_replica_weights(layer, i_replica: int): """ Get the weights for the given single replica ``i_replica``, - from a ``layer`` that has weights for all replicas. + from a ``layer`` that contains the weights of all the replicas. Note that the layer could be a complete NN with many separated sub_layers each of which containing weights for all replicas together.