| /third_party/mindspore/mindspore/train/ |
| D | model.py | 360 def _warmup_dataset(self, epoch, train_dataset, sink_size=-1): argument
385 def _init(self, train_dataset=None, valid_dataset=None, sink_size=-1, epoch=1): argument
442 def _train(self, epoch, train_dataset, callbacks=None, dataset_sink_mode=True, sink_size=-1): argument
517 …def _train_dataset_sink_process(self, epoch, train_dataset, list_callback=None, cb_params=None, si… argument
586 def _train_process(self, epoch, train_dataset, list_callback=None, cb_params=None): argument
652 def train(self, epoch, train_dataset, callbacks=None, dataset_sink_mode=True, sink_size=-1): argument
728 def build(self, train_dataset=None, valid_dataset=None, sink_size=-1, epoch=1): argument
940 def _infer_train_check(self, train_dataset, dataset_sink_mode, sink_size): argument
969 def infer_train_layout(self, train_dataset, dataset_sink_mode=True, sink_size=-1): argument
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | model_thor.py | 281 def init(self, train_dataset=None, valid_dataset=None): argument
365 def _train(self, epoch, train_dataset, callbacks=None, dataset_sink_mode=True): argument
410 def _train_dataset_sink_process(self, epoch, train_dataset, list_callback=None, cb_params=None): argument
475 def _train_process(self, epoch, train_dataset, list_callback=None, cb_params=None): argument
540 def train(self, epoch, train_dataset, callbacks=None, dataset_sink_mode=True): argument
|
| /third_party/mindspore/mindspore/nn/probability/infer/variational/ |
| D | svi.py | 52 def run(self, train_dataset, epochs=10): argument
|
| /third_party/mindspore/mindspore/train/train_thor/ |
| D | model_thor.py | 169 …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, si… argument
|
| /third_party/mindspore/mindspore/nn/probability/toolbox/ |
| D | anomaly_detection.py | 47 def train(self, train_dataset, epochs=5): argument
|
| D | uncertainty_evaluation.py | 85 def __init__(self, model, train_dataset, task_type, num_classes=None, epochs=1, argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/deeplabv3/ |
| D | train_one_epoch_with_loss.py | 76 train_dataset = create_dataset(args_opt, args_opt.data_url, 1, config.batch_size, variable
|
| /third_party/mindspore/tests/mindspore_test_framework/utils/ |
| D | model_util.py | 108 def train(self, num_epochs, train_dataset): argument
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | lineage.proto | 88 optional TrainDataset train_dataset = 2; field
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gradient_accumulation.py | 148 def train_process(self, epoch, train_dataset, mini_steps=None): argument
|