Home
last modified time | relevance | path

Searched refs:epoch_num (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/mindspore/train/train_thor/
Ddataset_helper.py25 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 …]
Dmodel_thor.py128 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/
Ddataset_helper.py27 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 …]
Dmodel.py334 …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/
Dcallback_param.h29 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/
Dtest_callback.py105 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/
Dbatch_op.cc101 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/
Drun_net_train.sh23 …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/
Dtest_getnext_dynamic_pipeline.py23 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/
Dmd_dataset.py81 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/
Dcache_perf_run.cc573 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/
Dutils.py80 cb_params.cur_epoch_num, cb_params.epoch_num, cur_step_in_epoch +
/third_party/mindspore/tests/st/quantization/mobilenetv2_quant/
Dutils.py82 cb_params.cur_epoch_num, cb_params.epoch_num, cur_step_in_epoch +
/third_party/mindspore/tests/ut/python/dataset/
Dtest_callbacks.py82 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/
Dtinybert_train_export.py154 epoch_num = 1 variable
/third_party/mindspore/mindspore/train/callback/
D_summary_collector.py400 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/
Dtransforms.py483 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/
Dtest_gradient_accumulation.py152 dataset_helper = DatasetHelper(train_dataset, dataset_sink_mode=False, epoch_num=epoch)
/third_party/mindspore/tests/st/networks/models/resnet50/src_thor/
Dmodel_thor.py390 cb_params.epoch_num = epoch