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

Method amp_params

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

Source from the content-addressed store, hash-verified

1313 return self._config.amp_enabled
1314
1315 def amp_params(self):
1316 return self._config.amp_params
1317
1318 def torch_autocast_enabled(self) -> bool:
1319 return self._config.torch_autocast_enabled

Callers 1

_configure_optimizerMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected