From cc0d9f72164f724f5883d0b7f353a602ca64d84e Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
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))