original_docstring: Returns dictionary of values necessary to reconstruct scheduler
docstring: Returns dictionary of values necessary to reconstruct scheduler
docstring_tokens: Returns dictionary of values necessary to reconstruct scheduler
code: def state_dict(self):
          """Returns dictionary of values necessary to reconstruct scheduler"""
          state_dict = {
              "warmup": self.warmup,
              "step": self._step,
              "decay_length": self.decay_length,
              "peak_rate": self.peak_rate,
              "rate": self._rate,
              "decay_rate": self.decay_rate,
              "min_rate": self.min_rate
          }
          return state_dict
code_tokens: def state_dict ( self ) : state_dict = { "warmup" : self . warmup , "step" : self . _step , "decay_length" : self . decay_length , "peak_rate" : self . peak_rate , "rate" : self . _rate , "decay_rate" : self . decay_rate , "min_rate" : self . min_rate } return state_dict
short_docstring: Returns dictionary of values necessary to reconstruct scheduler
short_docstring_tokens: Returns dictionary of values necessary to reconstruct scheduler
comment: """Returns dictionary of values necessary to reconstruct scheduler"""