Searched refs:multisteplr (Results 1 – 3 of 3) sorted by relevance
/third_party/mindspore/tests/ut/python/nn/optim/ |
D | test_lars.py | 27 def multisteplr(total_steps, milestone, base_lr=0.9, gamma=0.1, dtype=mstype.float32): function 57 lr = multisteplr(10, [2, 6])
|
/third_party/mindspore/tests/st/nccl/ |
D | test_nccl_lenet.py | 72 def multisteplr(total_steps, gap, base_lr=0.9, gamma=0.1, dtype=mstype.float32): function 85 learning_rate = multisteplr(epoch, 2)
|
/third_party/mindspore/tests/st/networks/ |
D | test_gpu_lenet.py | 122 def multisteplr(total_steps, gap, base_lr=0.9, gamma=0.1, dtype=mstype.float32): function 137 learning_rate = multisteplr(epoch, 30)
|