MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeed / _get_optimizer_param

Method _get_optimizer_param

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

Source from the content-addressed store, hash-verified

3435 self.monitor.write_events(self.summary_events)
3436
3437 def _get_optimizer_param(self, param_name):
3438 result = []
3439 if not self.optimizer:
3440 return result
3441 for group in self.optimizer.param_groups:
3442 if param_name in group:
3443 result.append(group[param_name])
3444 else:
3445 result.append(0.0)
3446 return result
3447
3448 def _get_optimizer_loss_scale(self):
3449 if not self.optimizer:

Callers 3

get_lrMethod · 0.95
get_typeMethod · 0.95
get_momMethod · 0.95

Calls 1

appendMethod · 0.80

Tested by

no test coverage detected