Home
last modified time | relevance | path

Searched refs:ModeEnum (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/train/callback/
D_summary_collector.py32 from mindspore.train.summary.enums import PluginEnum, ModeEnum
360 if cb_params.mode not in ModeEnum.to_list():
369 if cb_params.mode != ModeEnum.TRAIN.value:
427 if cb_params.mode == ModeEnum.TRAIN.value:
523 if cb_params.mode == ModeEnum.TRAIN.value:
536 …network = cb_params.train_network if cb_params.mode == ModeEnum.TRAIN.value else cb_params.eval_ne…
791 …dataset = cb_params.train_dataset if cb_params.mode == ModeEnum.TRAIN.value else cb_params.valid_d…
809 if cb_params.mode == ModeEnum.TRAIN.value:
894 if cb_params.mode == ModeEnum.TRAIN.value:
/third_party/mindspore/tests/ut/python/train/summary/
Dtest_summary_collector.py29 from mindspore.train.summary.enums import ModeEnum, PluginEnum
338 cb_params.mode = ModeEnum.TRAIN.value
383 @pytest.mark.parametrize('mode', [ModeEnum.TRAIN.value, ModeEnum.EVAL.value])
389 if mode == ModeEnum.TRAIN.value:
412 cb_params.mode = ModeEnum.TRAIN.value
/third_party/mindspore/mindspore/train/summary/
Denums.py50 class ModeEnum(BaseEnum): class