MCPcopy Create free account
hub / github.com/AIS-SNU/Smart-Infinity / train_batch_size

Method train_batch_size

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

Source from the content-addressed store, hash-verified

630 return self._config.sparse_gradients_enabled
631
632 def train_batch_size(self):
633 return self._config.train_batch_size
634
635 def train_micro_batch_size_per_gpu(self):
636 return self._config.train_micro_batch_size_per_gpu

Callers 6

__init__Method · 0.95
_take_model_stepMethod · 0.95
_autotuning_exitMethod · 0.95
_load_checkpointMethod · 0.95
__init__Method · 0.80
train_batchMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected