Home
last modified time | relevance | path

Searched defs:train_dataset (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/train/
Dmodel.py628 def _warmup_dataset(self, epoch, train_dataset, sink_size=-1): argument
653 def _waiting_for_dataset_warmup_ready(self, train_dataset): argument
669 def _init(self, train_dataset=None, valid_dataset=None, sink_size=-1, epoch=1): argument
750 …def _train(self, epoch, train_dataset, callbacks=None, dataset_sink_mode=True, sink_size=-1, initi… argument
822 def _train_dataset_sink_process(self, epoch, train_dataset, list_callback=None, cb_params=None, argument
1066 …def _train_process(self, epoch, train_dataset, list_callback=None, cb_params=None, initial_epoch=0, argument
1178 …def train(self, epoch, train_dataset, callbacks=None, dataset_sink_mode=False, sink_size=-1, initi… argument
1336 def fit(self, epoch, train_dataset, valid_dataset=None, valid_frequency=1, callbacks=None, argument
1453 def build(self, train_dataset=None, valid_dataset=None, sink_size=-1, epoch=1): argument
1968 def _infer_train_check(self, train_dataset, dataset_sink_mode, sink_size): argument
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/train/train_thor/
Dmodel_thor.py169 …def _train_ascend_sink_step(self, cb_params, train_dataset, iter_first_order, inputs, list_callbac… argument
190 def _train_dataset_sink_process(self, epoch, train_dataset, list_callback=None, cb_params=None, argument
/third_party/mindspore/mindspore-src/source/tests/mindspore_test_framework/utils/
Dmodel_util.py108 def train(self, num_epochs, train_dataset): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dlineage.proto88 optional TrainDataset train_dataset = 2; field
/third_party/mindspore/mindspore-src/source/tests/st/networks/
Dtest_gradient_accumulation.py145 def train_process(self, epoch, train_dataset, mini_steps=None): argument