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.warmup = state_dict['warmup']
          self._step = state_dict['step']
          self.decay_length = state_dict['decay_length']
          self.peak_rate = state_dict['peak_rate']
          self._rate = state_dict['rate']
          self.decay_rate = state_dict['decay_rate']
          self.min_rate = state_dict['min_rate']
code_tokens: def load_state_dict ( self , state_dict ) : self . warmup = state_dict [ 'warmup' ] self . _step = state_dict [ 'step' ] self . decay_length = state_dict [ 'decay_length' ] self . peak_rate = state_dict [ 'peak_rate' ] self . _rate = state_dict [ 'rate' ] self . decay_rate = state_dict [ 'decay_rate' ] self . min_rate = state_dict [ 'min_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"""