From 1881d106fb39538e5477f851b13fea1694c67b2e Mon Sep 17 00:00:00 2001 From: Lenz Fiedler Date: Tue, 29 Oct 2024 14:57:48 +0100 Subject: [PATCH] Corrected one line only to have the second one be incorrect too --- mala/network/runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mala/network/runner.py b/mala/network/runner.py index a8910c2ad..ff111b10b 100644 --- a/mala/network/runner.py +++ b/mala/network/runner.py @@ -852,7 +852,7 @@ def _forward_entire_snapshot( # activated AFTER loading a model. if self.parameters_full.use_ddp: self.network.module.to( - self.network.params._configuration["device"] + self.network.module.params._configuration["device"] ) else: self.network.to(self.network.params._configuration["device"])