From cc0d9f72164f724f5883d0b7f353a602ca64d84e Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 24 Jan 2022 18:05:45 -0700 Subject: [PATCH] Fix --- codes/trainer/base_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/trainer/base_model.py b/codes/trainer/base_model.py index 56f33603..4ea7edae 100644 --- a/codes/trainer/base_model.py +++ b/codes/trainer/base_model.py @@ -129,7 +129,7 @@ class BaseModel(): network.load_state_dict(load_net_clean, strict=strict) - def consolidate_state(self, state): + def consolidate_state(self): for o in self.optimizers: if isinstance(o, ZeroRedundancyOptimizer): state['optimizers'].append(o.consolidate_state_dict(to=0))