MCPcopy Create free account
hub / github.com/AIS-SNU/Smart-Infinity / _get_optimizer_param

Method _get_optimizer_param

deepspeed/runtime/engine.py:2190–2199  ·  view source on GitHub ↗
(self, param_name)

Source from the content-addressed store, hash-verified

2188 self.monitor.write_events(self.summary_events)
2189
2190 def _get_optimizer_param(self, param_name):
2191 result = []
2192 if not self.optimizer:
2193 return result
2194 for group in self.optimizer.param_groups:
2195 if param_name in group:
2196 result.append(group[param_name])
2197 else:
2198 result.append(0.0)
2199 return result
2200
2201 def get_lr(self):
2202 return self._get_optimizer_param("lr")

Callers 3

get_lrMethod · 0.95
get_typeMethod · 0.95
get_momMethod · 0.95

Calls 1

appendMethod · 0.45

Tested by

no test coverage detected