Home
last modified time | relevance | path

Searched defs:micro_batch_num (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/parallel/nn/
Dop_parallel_config.py96 def __init__(self, pipeline_stage=1, micro_batch_num=1): argument
113 def micro_batch_num(self): member in _PipeLineConfig
117 def micro_batch_num(self, value): member in _PipeLineConfig
Dtransformer.py144 …def __init__(self, data_parallel=1, model_parallel=1, pipeline_stage=1, micro_batch_num=1, recompu… argument
180 def micro_batch_num(self): member in TransformerOpParallelConfig
184 def micro_batch_num(self, value): member in TransformerOpParallelConfig