diff --git a/deepspeed/runtime/engine.py b/deepspeed/runtime/engine.py index 82a7592f14cb..635b38a8808a 100755 --- a/deepspeed/runtime/engine.py +++ b/deepspeed/runtime/engine.py @@ -2965,11 +2965,11 @@ def step(self, lr_kwargs=None): if (self.eigenvalue_enabled() and not self.gas_boundary_ctr % self.eigenvalue_gas_boundary_resolution()): - ev_values = self.block_eigenvalue.values() + ev_values = list(self.block_eigenvalue.values()) for i in range(len(ev_values)): self.summary_events.append(( f"Train/Eigenvalues/ModelBlockParam_{i}", - self.ev_values[i][0], + ev_values[i][0], self.global_samples, )) self.monitor.write_events(self.summary_events)