Searched defs:lr_init (Results 1 – 6 of 6) sorted by relevance
| /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
|
| D | utils.py | 43 def __init__(self, lr_init=None, step_threshold=10): argument
|
| /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
|
| /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
|
| D | utils.py | 43 def __init__(self, lr_init=None, step_threshold=10): argument
|
| /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
|