| /third_party/mindspore/mindspore/train/ |
| D | model.py | 360 def _warmup_dataset(self, epoch, train_dataset, sink_size=-1): argument
373 epoch_num = math.ceil(epoch * sink_size / train_dataset.get_dataset_size())
374 train_dataset.__total_batch__ = epoch * sink_size
377 dataset=train_dataset,
382 train_dataset._dataset_helper = dataset_helper
383 train_dataset._warmup_epoch = epoch
385 def _init(self, train_dataset=None, valid_dataset=None, sink_size=-1, epoch=1): argument
403 if not train_dataset and not valid_dataset:
408 if train_dataset:
413 train_dataset.__no_send__ = True
[all …]
|
| /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
308 if not train_dataset and not valid_dataset:
313 if train_dataset:
322 train_dataset.__loop_size__ = iter_second_order
326 dataset=train_dataset,
340 _exec_datagraph(train_dataset, iter_first_order, phase='train1_dataset')
365 def _train(self, epoch, train_dataset, callbacks=None, dataset_sink_mode=True): argument
391 cb_params.batch_num = train_dataset.get_dataset_size()
397 cb_params.train_dataset = train_dataset
402 self._train_process(epoch, train_dataset, list_callback, cb_params)
[all …]
|
| /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
183 _exec_datagraph(train_dataset, iter_first_order, phase='train1_dataset')
190 …def _train_dataset_sink_process(self, epoch, train_dataset, list_callback=None, cb_params=None, si… argument
208 epoch_num = math.ceil(epoch * sink_size / train_dataset.get_dataset_size())
212 train_dataset.__loop_size__ = iter_second_order
216 dataset=train_dataset,
240 … self._train_ascend_sink_step(cb_params, train_dataset, iter_first_order, inputs, list_callback,
|
| /third_party/mindspore/tests/mindspore_test_framework/utils/ |
| D | model_util.py | 108 def train(self, num_epochs, train_dataset): argument
109 train_dataset = list(train_dataset)
111 for x, y in train_dataset:
|
| /third_party/mindspore/tests/mindspore_test_framework/components/executor/ |
| D | exec_and_verify_model_loss.py | 43 train_dataset = self.inputs[keyword.desc_inputs]
45 loss = model.train(num_epochs, train_dataset)
|
| /third_party/mindspore/tests/st/networks/models/deeplabv3/ |
| D | test_deeplabv3.py | 83 train_dataset = create_dataset(args_opt, data_url, 1, config.batch_size,
85 dataset_size = train_dataset.get_dataset_size()
96 model.train(epoch_size, train_dataset, callback)
|
| /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
78 dataset_size = train_dataset.get_dataset_size()
89 model.train(epoch_size, train_dataset, callback)
|
| /third_party/mindspore/mindspore/nn/probability/toolbox/ |
| D | anomaly_detection.py | 47 def train(self, train_dataset, epochs=5): argument
62 self.vae = vi.run(train_dataset, epochs)
|
| D | uncertainty_evaluation.py | 85 def __init__(self, model, train_dataset, task_type, num_classes=None, epochs=1, argument
89 self.epi_train_dataset = train_dataset
90 self.ale_train_dataset = deepcopy(train_dataset)
|
| /third_party/mindspore/mindspore/nn/probability/infer/variational/ |
| D | svi.py | 52 def run(self, train_dataset, epochs=10): argument
69 for data in train_dataset.create_dict_iterator(num_epochs=1):
|
| /third_party/mindspore/mindspore/train/callback/ |
| D | _summary_collector.py | 524 train_dataset = cb_params.train_dataset
526 graph_bytes = dataset_graph.package_dataset_graph(train_dataset)
755 …lineage_message.train_dataset.train_dataset_path = train_lineage.get(LineageMetadata.train_dataset…
757 …lineage_message.train_dataset.train_dataset_size = train_lineage.get(LineageMetadata.train_dataset…
791 …dataset = cb_params.train_dataset if cb_params.mode == ModeEnum.TRAIN.value else cb_params.valid_d…
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gradient_accumulation.py | 148 def train_process(self, epoch, train_dataset, mini_steps=None): argument
152 dataset_helper = DatasetHelper(train_dataset, dataset_sink_mode=False, epoch_num=epoch)
164 train_dataset.reset()
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_bias_add.py | 94 model.train(epoch=1, train_dataset=dataset, dataset_sink_mode=False)
|
| /third_party/mindspore/tests/st/probability/dpn/ |
| D | test_gpu_svi_vae.py | 107 vae = vi.run(train_dataset=ds_train, epochs=5)
|
| D | test_gpu_svi_cvae.py | 118 cvae = vi.run(train_dataset=ds_train, epochs=5)
|
| D | test_gpu_vae_gan.py | 165 vae_gan = vi.run(train_dataset=ds_train, epochs=5)
|
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_cache_nomap.py | 1445 train_dataset = ds.TFRecordDataset(TRAIN_DATA_DIR, TRAIN_SCHEMA_DIR)
1447 …train_dataset = train_dataset.map(input_columns=["image"], operations=decode_op, cache=train_cache)
1454 train_iter = train_dataset.create_dict_iterator(num_epochs=num_epoch)
1578 train_dataset = ds.TFRecordDataset(TRAIN_DATA_DIR, TRAIN_SCHEMA_DIR)
1580 …train_dataset = train_dataset.map(input_columns=["image"], operations=decode_op, cache=train_cache)
1583 train_iter = train_dataset.create_dict_iterator(num_epochs=num_epoch)
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | lineage.proto | 88 optional TrainDataset train_dataset = 2; field
|
| /third_party/mindspore/tests/st/probability/toolbox/ |
| D | test_uncertainty.py | 125 train_dataset=ds_train,
|
| /third_party/mindspore/tests/ut/python/communication/ |
| D | test_data_parallel_resnet.py | 292 model.train(epoch=2, train_dataset=dataset, dataset_sink_mode=False)
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | optimizer_parallel.py | 268 train_dataset=dataset,
|
| D | multifieldembeddinglookup_parallel.py | 258 train_dataset=dataset,
|
| D | parallel_strategy_search.py | 293 train_dataset=dataset,
|
| /third_party/mindspore/tests/ut/python/train/summary/ |
| D | test_summary_collector.py | 337 …cb_params.train_dataset = dataset.MnistDataset(dataset_dir=tempfile.mkdtemp(dir=self.base_summary_…
|
| /third_party/mindspore/mindspore/nn/probability/ |
| D | README.md | 237 vae = vi.run(train_dataset=ds_train, epochs=10)
453 train_dataset=ds_train,
|