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

Method train_batch_size

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

Source from the content-addressed store, hash-verified

1072 return self._config.sparse_gradients_enabled
1073
1074 def train_batch_size(self):
1075 return self._config.train_batch_size
1076
1077 def train_micro_batch_size_per_gpu(self):
1078 return self._config.train_micro_batch_size_per_gpu

Callers 9

__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
get_dataloader_and_lr_scheduler_for_variable_batch_size_deepspeedFunction · 0.80
_take_lr_scheduler_stepFunction · 0.80
mainFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected