/third_party/mindspore/mindspore/train/train_thor/ |
D | dataset_helper.py | 25 def _send_data(dataset, epoch_num): argument 29 exec_dataset.send(epoch_num) 33 def _send_data_no_flag(dataset, epoch_num): argument 36 exec_dataset.send(epoch_num) 64 …def __init__(self, dataset, dataset_sink_mode=True, sink_size=-1, epoch_num=1, iter_first_order=1): argument 73 self.iter = iterclass(dataset, sink_size, epoch_num, iter_first_order) 76 self.iter = iterclass(dataset, sink_size, epoch_num) 99 def __init__(self, dataset, sink_size, epoch_num): argument 110 _send_data(dataset, epoch_num) 112 _send_data_no_flag(dataset, epoch_num) [all …]
|
D | model_thor.py | 128 epoch_num=1, iter_first_order=1): argument 132 … dataset_helper = DatasetHelper(dataset, dataset_sink_mode, sink_size, epoch_num, iter_first_order) 206 epoch_num = epoch 208 epoch_num = math.ceil(epoch * sink_size / train_dataset.get_dataset_size()) 219 epoch_num=epoch_num,
|
/third_party/mindspore/mindspore/train/ |
D | dataset_helper.py | 27 def _send_data(dataset, epoch_num): argument 31 exec_dataset.send(epoch_num) 35 def _send_data_no_flag(dataset, epoch_num): argument 38 exec_dataset.send(epoch_num) 225 def __init__(self, dataset, dataset_sink_mode=True, sink_size=-1, epoch_num=1): argument 251 self.iter = iterclass(dataset, sink_size, epoch_num) 254 self.iter = iterclass(dataset, epoch_num=epoch_num) 336 def __init__(self, dataset, sink_size, epoch_num): argument 353 _send_data(dataset, epoch_num) 355 _send_data_no_flag(dataset, epoch_num) [all …]
|
D | model.py | 334 …def _exec_preprocess(self, is_train, dataset, dataset_sink_mode, sink_size=-1, epoch_num=1, datase… argument 347 dataset_helper = DatasetHelper(dataset, dataset_sink_mode, sink_size, epoch_num) 371 epoch_num = epoch 373 epoch_num = math.ceil(epoch * sink_size / train_dataset.get_dataset_size()) 380 epoch_num=epoch_num, 474 cb_params.epoch_num = epoch 534 epoch_num = epoch 536 epoch_num = math.ceil(epoch * sink_size / train_dataset.get_dataset_size()) 556 epoch_num=epoch_num, 603 epoch_num=epoch)
|
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/callback/ |
D | callback_param.h | 29 CallbackParam(int64_t epoch_num, int64_t cur_epoch_step, int64_t total_step_num) in CallbackParam() argument 30 … : cur_epoch_num_(epoch_num), cur_epoch_step_num_(cur_epoch_step), cur_step_num_(total_step_num) {} in CallbackParam()
|
/third_party/mindspore/tests/ut/python/utils/ |
D | test_callback.py | 105 cb_params.epoch_num = 10 122 cb_params.epoch_num = 4 144 cb_params.epoch_num = 4 208 cb_params.epoch_num = 10 237 cb_params.epoch_num = 10 268 cb_params.epoch_num = 10 380 cb_params.epoch_num = 10
|
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
D | batch_op.cc | 101 int64_t epoch_num = 0, batch_num = 0, cnt = 0; in operator ()() local 114 … std::make_pair(std::move(table), CBatchInfo(epoch_num, batch_num++, cnt + 1 - epoch_num)))); in operator ()() 117 …RETURN_IF_NOT_OK(GetBatchSize(&cur_batch_size, CBatchInfo(epoch_num, batch_num, cnt - epoch_num))); in operator ()() 124 std::make_pair(std::move(table), CBatchInfo(epoch_num, batch_num++, cnt + 1 - epoch_num)))); in operator ()() 130 epoch_num++; in operator ()() 133 …RETURN_IF_NOT_OK(GetBatchSize(&cur_batch_size, CBatchInfo(epoch_num, batch_num, cnt - epoch_num))); in operator ()()
|
/third_party/mindspore/mindspore/lite/test/st/scripts/ |
D | run_net_train.sh | 23 …vileged=true "${docker_image}" python ${models_path}'/'${model_name}_train_export.py "${epoch_num}" 26 …D_MODEL_ZOO=${CLOUD_MODEL_ZOO} python ${models_path}'/'${model_name}_train_export.py "${epoch_num}" 231 …--expectedDataFile=${train_io_path}/${model_prefix}_output --epochs=${epoch_num} --numThreads=${th… 241 …--expectedDataFile=${train_io_path}/${model_prefix}_output --epochs=${epoch_num} --numThreads=${th… 365 --epochs=${epoch_num} \ 486 epoch_num=1 529 epoch_num=${OPTARG} 530 echo "train epoch num is ${epoch_num}"
|
/third_party/mindspore/tests/st/dynamic_shape/ |
D | test_getnext_dynamic_pipeline.py | 23 def _exec_preprocess(network, is_train, dataset, dataset_sink_mode, sink_size=-1, epoch_num=1, data… argument 28 dataset_helper = DatasetHelper(dataset, dataset_sink_mode, sink_size, epoch_num)
|
/third_party/mindspore/tests/st/networks/models/deeplabv3/src/ |
D | md_dataset.py | 81 def create_dataset(args, data_url, epoch_num=1, batch_size=1, usage="train", shuffle=True): argument 111 dataset = dataset.repeat(count=epoch_num)
|
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/cache/perf/ |
D | cache_perf_run.cc | 573 auto epoch_num = 2; in Run() local 574 while (epoch_num <= num_epoches_) { in Run() 589 … std::cout << "Epoch " << epoch_num << " elapsed time " << elapse_time << " seconds" << std::endl; in Run() 590 std::cout << "Epoch " << epoch_num in Run() 593 ++epoch_num; in Run()
|
/third_party/mindspore/tests/st/quantization/resnet50_quant/ |
D | utils.py | 80 cb_params.cur_epoch_num, cb_params.epoch_num, cur_step_in_epoch +
|
/third_party/mindspore/tests/st/quantization/mobilenetv2_quant/ |
D | utils.py | 82 cb_params.cur_epoch_num, cb_params.epoch_num, cur_step_in_epoch +
|
/third_party/mindspore/tests/ut/python/dataset/ |
D | test_callbacks.py | 82 def generate_expected(epoch_num, step_num, step_size=1, map_num=1, repeat=1): argument 92 for e in range(epoch_num):
|
/third_party/mindspore/mindspore/lite/examples/export_models/models/ |
D | tinybert_train_export.py | 154 epoch_num = 1 variable
|
/third_party/mindspore/mindspore/train/callback/ |
D | _summary_collector.py | 400 total_step = cb_params.epoch_num 718 train_lineage[LineageMetadata.epoch] = cb_params.epoch_num
|
/third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
D | transforms.py | 483 epoch_num = batchInfo.get_epoch_num() 485 seed_key = self.seed_list[(epoch_num * self.resize_count_num + size_idx) % self.seed_num]
|
/third_party/mindspore/tests/st/networks/ |
D | test_gradient_accumulation.py | 152 dataset_helper = DatasetHelper(train_dataset, dataset_sink_mode=False, epoch_num=epoch)
|
/third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
D | model_thor.py | 390 cb_params.epoch_num = epoch
|