original_docstring: Given a state_dict, this function loads scheduler's state
docstring: Given a state_dict, this function loads scheduler's state
docstring_tokens: Given a state_dict this function loads scheduler ' s state
code: def load_state_dict(self, state_dict):
          """Given a state_dict, this function loads scheduler's state"""
          self._step = state_dict["step"]
          self.warmup = state_dict["warmup"]
          self.factor = state_dict["factor"]
          self.hidden_size = state_dict["hidden_size"]
          self._rate = state_dict["rate"]
code_tokens: def load_state_dict ( self , state_dict ) : self . _step = state_dict [ "step" ] self . warmup = state_dict [ "warmup" ] self . factor = state_dict [ "factor" ] self . hidden_size = state_dict [ "hidden_size" ] self . _rate = state_dict [ "rate" ]
short_docstring: Given a state_dict, this function loads scheduler's state
short_docstring_tokens: Given a state_dict this function loads scheduler ' s state
comment: """Given a state_dict, this function loads scheduler's state"""