From abd5f123b8f34f92a4362fb51409969f447e8c6e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Lucas=20de=20Sousa=20Almeida?= Date: Fri, 21 Jul 2023 10:14:15 -0300 Subject: [PATCH] kwargs for the sub-autoencoders of MultiScaleAutoencoder MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: João Lucas de Sousa Almeida --- simulai/models/_pytorch_models/_autoencoder.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/simulai/models/_pytorch_models/_autoencoder.py b/simulai/models/_pytorch_models/_autoencoder.py index fce211d5..0b855262 100644 --- a/simulai/models/_pytorch_models/_autoencoder.py +++ b/simulai/models/_pytorch_models/_autoencoder.py @@ -1779,7 +1779,8 @@ def __init__(self, shallow:bool=True, scale: float = 1e-3, devices:Union[str, List]="cpu", - name:str=None) -> None: + name:str=None, + **kwargs) -> None: super(MultiScaleAutoencoder, self).__init__(name=name) @@ -1805,6 +1806,7 @@ def __init__(self, devices=devices, shallow=shallow, name=name, + **kwargs, ) self.AutoencodersList.append(autoencoder)