Skip to content

Commit b3fc750

Browse files
committed
fix typo
1 parent 75f7d2d commit b3fc750

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

trinity/algorithm/sample_strategy/mix_sample_strategy.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -114,12 +114,12 @@ def default_args(cls) -> Dict:
114114

115115
def state_dict(self) -> dict:
116116
return {
117-
"usal_buffer": self.usual_exp_buffer.state_dict(),
117+
"usual_buffer": self.usual_exp_buffer.state_dict(),
118118
"expert_buffer": self.expert_exp_buffer.state_dict(),
119119
}
120120

121121
def load_state_dict(self, state_dict: dict) -> None:
122-
if state_dict.get("usal_buffer", None):
123-
self.usual_exp_buffer.load_state_dict(state_dict["usal_buffer"])
122+
if state_dict.get("usual_buffer", None):
123+
self.usual_exp_buffer.load_state_dict(state_dict["usual_buffer"])
124124
if state_dict.get("expert_buffer", None):
125125
self.expert_exp_buffer.load_state_dict(state_dict["expert_buffer"])

0 commit comments

Comments
 (0)