Method zero_optimization

deepspeed/runtime/engine.py:668–669 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

666	)
667
668	def zero_optimization(self):
669	return self._config.zero_enabled
670
671	def zero_allow_untested_optimizer(self):
672	return self._config.zero_allow_untested_optimizer

__init__Method · 0.95

get_data_typesMethod · 0.95

_configure_checkpointingMethod · 0.95

_do_optimizer_sanity_checkMethod · 0.95

_configure_basic_optimizerMethod · 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

_get_zero_param_shapesMethod · 0.95

no outgoing calls

no test coverage detected