Unverified Commit 125d7830 authored by Ayan Das's avatar Ayan Das Committed by GitHub
Browse files

fix typo in EMAModel's load_state_dict() (#2151)

Possible typo introduced in 7c82a16f
parent fdf70cb5
......@@ -253,7 +253,7 @@ class EMAModel:
if not isinstance(self.inv_gamma, (float, int)):
raise ValueError("Invalid inv_gamma")
self.power = state_dict["power"].get("power", self.power)
self.power = state_dict.get("power", self.power)
if not isinstance(self.power, (float, int)):
raise ValueError("Invalid power")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment