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

Method amp_enabled

deepspeed/runtime/engine.py:1326–1327  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

1324 return self._config.bfloat16_config.bf16_optimizer_states
1325
1326 def amp_enabled(self):
1327 return self._config.amp_enabled
1328
1329 def amp_params(self):
1330 return self._config.amp_params

Callers 8

_do_sanity_checkMethod · 0.95
_configure_distributed_modelMethod · 0.95
_do_optimizer_sanity_checkMethod · 0.95
_backward_prologueMethod · 0.95
_backward_post_hookMethod · 0.95
scaleMethod · 0.95
backwardMethod · 0.95
_take_model_stepMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected