From 35959881b2af339f8266533544450d79adafdaa7 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Tue, 9 Jan 2024 13:17:24 +0100 Subject: [PATCH] Fix (scaling/standalone): add return statement in state_dict (#792) --- src/brevitas/core/scaling/standalone.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/brevitas/core/scaling/standalone.py b/src/brevitas/core/scaling/standalone.py index eb8f7778f..03d84a8a2 100644 --- a/src/brevitas/core/scaling/standalone.py +++ b/src/brevitas/core/scaling/standalone.py @@ -223,6 +223,7 @@ def state_dict(self, destination=None, prefix='', keep_vars=False): # Avoid saving the init value if not self.init_done: del output_dict[prefix + 'value'] + return output_dict def _load_from_state_dict( self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys,