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

Method amp_params

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

Source from the content-addressed store, hash-verified

1259 return self._config.amp_enabled
1260
1261 def amp_params(self):
1262 return self._config.amp_params
1263
1264 def torch_autocast_enabled(self) -> bool:
1265 return self._config.torch_autocast_enabled

Callers 1

_configure_optimizerMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected