File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
trinity/algorithm/sample_strategy Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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" ])
You can’t perform that action at this time.
0 commit comments