Method zero_optimization

deepspeed/runtime/engine.py:1052–1053 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

1050	)
1051
1052	def zero_optimization(self):
1053	return self._config.zero_enabled
1054
1055	def zero_allow_untested_optimizer(self):
1056	return self._config.zero_allow_untested_optimizer

__init__Method · 0.95

_do_optimizer_sanity_checkMethod · 0.95

_configure_basic_optimizerMethod · 0.95

_backward_prologueMethod · 0.95

backwardMethod · 0.95

_take_model_stepMethod · 0.95

load_checkpointMethod · 0.95

_load_checkpointMethod · 0.95

_save_moe_checkpointMethod · 0.95

_save_checkpointMethod · 0.95

_save_zero_checkpointMethod · 0.95

no outgoing calls

no test coverage detected