| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | lr_scheduler.py | 29 def warmup_step_lr(lr, lr_epochs, steps_per_epoch, warmup_epochs, max_epoch, gamma=0.1): argument
33 total_steps = int(max_epoch * steps_per_epoch)
34 warmup_steps = int(warmup_epochs * steps_per_epoch)
38 milestones_step = milestone * steps_per_epoch
54 def multi_step_lr(lr, milestones, steps_per_epoch, max_epoch, gamma=0.1): argument
55 return warmup_step_lr(lr, milestones, steps_per_epoch, 0, max_epoch, gamma=gamma)
58 def step_lr(lr, epoch_size, steps_per_epoch, max_epoch, gamma=0.1): argument
63 return multi_step_lr(lr, lr_epochs, steps_per_epoch, max_epoch, gamma=gamma)
66 def warmup_cosine_annealing_lr(lr, steps_per_epoch, warmup_epochs, max_epoch, T_max, eta_min=0): argument
70 total_steps = int(max_epoch * steps_per_epoch)
[all …]
|
| /third_party/mindspore/tests/st/quantization/mobilenetv2_quant/ |
| D | lr_generator.py | 21 def get_lr(global_step, lr_init, lr_end, lr_max, warmup_epochs, total_epochs, steps_per_epoch): argument
38 total_steps = steps_per_epoch * total_epochs
39 warmup_steps = steps_per_epoch * warmup_epochs
|
| D | test_mobilenetv2_quant_gpu.py | 96 steps_per_epoch=step_size))
|
| D | test_mobilenetv2_quant.py | 98 steps_per_epoch=step_size))
|
| /third_party/mindspore/tests/st/quantization/resnet50_quant/ |
| D | lr_generator.py | 21 def get_lr(lr_init, lr_end, lr_max, warmup_epochs, total_epochs, steps_per_epoch, lr_decay_mode): argument
38 total_steps = steps_per_epoch * total_epochs
39 warmup_steps = steps_per_epoch * warmup_epochs
|
| D | test_resnet50_quant.py | 103 steps_per_epoch=step_size,
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src/ |
| D | lr_generator.py | 20 def get_learning_rate(lr_init, lr_end, lr_max, warmup_epochs, total_epochs, steps_per_epoch, lr_dec… argument
37 total_steps = steps_per_epoch * total_epochs
38 warmup_steps = steps_per_epoch * warmup_epochs
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/ |
| D | test_yolov3_darknet53.py | 146 steps_per_epoch = int(data_size / per_batch_size / group_size)
147 print("steps_per_epoch:", steps_per_epoch)
154 steps_per_epoch,
189 if (i + 1) % steps_per_epoch == 0:
191 epoch = int(i / steps_per_epoch)
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/ |
| D | lr_schedule.py | 28 def dynamic_lr(config, steps_per_epoch): argument
31 total_steps = steps_per_epoch * (config.epoch_size + 1)
|
| /third_party/mindspore/tests/st/networks/models/resnet50/ |
| D | test_resnet50_imagenet.py | 57 def get_thor_lr(global_step, lr_init, decay, total_epochs, steps_per_epoch, decay_epochs=100): argument
60 total_steps = steps_per_epoch * total_epochs
62 epoch = (i + 1) / steps_per_epoch
76 def get_thor_damping(global_step, damping_init, decay_rate, total_epochs, steps_per_epoch): argument
79 total_steps = steps_per_epoch * total_epochs
81 epoch = (step + 1) / steps_per_epoch
177 steps_per_epoch=step_size, lr_decay_mode=config.lr_decay_mode))
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/dump/ |
| D | data_dumper.cc | 193 void *steps_per_epoch = steps_per_epoch_tensor->device_address()->GetMutablePtr(); in SetOpMappingInfo() local
195 if (current_epoch != nullptr && current_step != nullptr && steps_per_epoch != nullptr) { in SetOpMappingInfo()
198 dump_info->set_iterations_per_loop_addr(reinterpret_cast<uint64_t>(steps_per_epoch)); in SetOpMappingInfo()
|