Method amp_enabled

deepspeed/runtime/engine.py:1258–1259 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

1256	return self._config.bfloat16_config.bf16_optimizer_states
1257
1258	def amp_enabled(self):
1259	return self._config.amp_enabled
1260
1261	def amp_params(self):
1262	return self._config.amp_params

_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

no outgoing calls

no test coverage detected