| /third_party/mindspore/mindspore/nn/ |
| D | __init__.py | 20 from . import layer, loss, optim, metrics, wrap, grad, probability, sparse, dynamic_lr
26 from .optim import *
36 __all__.extend(optim.__all__)
|
| /third_party/mindspore/tests/ut/python/utils/ |
| D | test_callback.py | 30 from mindspore.nn.optim import Momentum
101 optim = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
103 _train_network = TrainOneStepCell(network_, optim)
204 optim = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
206 _train_network = TrainOneStepCell(network_, optim)
233 optim = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
235 _train_network = TrainOneStepCell(network_, optim)
264 optim = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
266 _train_network = TrainOneStepCell(network_, optim)
|
| /third_party/ffmpeg/libavcodec/ |
| D | g723_1enc.c | 825 static void get_fcb_param(FCBParam *optim, int16_t *impulse_resp, in get_fcb_param() argument
959 if (err < optim->min_err) { in get_fcb_param()
960 optim->min_err = err; in get_fcb_param()
961 optim->grid_index = i; in get_fcb_param()
962 optim->amp_index = param.amp_index; in get_fcb_param()
963 optim->dirac_train = param.dirac_train; in get_fcb_param()
966 optim->pulse_sign[k] = param.pulse_sign[k]; in get_fcb_param()
967 optim->pulse_pos[k] = param.pulse_pos[k]; in get_fcb_param()
980 static void pack_fcb_param(G723_1_Subframe *subfrm, FCBParam *optim, in pack_fcb_param() argument
991 int val = buf[optim->grid_index + (i << 1)]; in pack_fcb_param()
[all …]
|
| /third_party/mindspore/tests/st/summary/ |
| D | test_summary_ops.py | 26 from mindspore.nn.optim import Momentum
106 optim = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
107 model = Model(net, loss_fn=loss, optimizer=optim, metrics={'loss': Loss()})
|
| D | test_summary_collector.py | 27 from mindspore.nn.optim import Momentum
113 optim = Momentum(lenet.trainable_params(), learning_rate=0.1, momentum=0.9)
114 model = Model(lenet, loss_fn=loss, optimizer=optim, metrics={'loss': Loss()})
|
| /third_party/mindspore/tests/ut/python/train/summary/ |
| D | test_image_summary.py | 23 from mindspore.nn.optim import Momentum
126 optim = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
128 model = Model(net, loss_fn=loss, optimizer=optim, metrics=None)
|
| /third_party/mindspore/mindspore/nn/probability/transforms/ |
| D | transform_bnn.py | 18 from ....nn import optim
222 modules = optim.__all__
227 optimizer = getattr(optim, name)
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_auto_monad_layer.py | 67 optim = nn.RMSProp(params=ae_with_loss.trainable_params(), learning_rate=0.002,)
68 train_net = nn.TrainOneStepCell(ae_with_loss, optim)
|
| /third_party/mindspore/tests/st/profiler/ |
| D | test_profiler.py | 26 from mindspore.nn.optim import Momentum
185 optim = Momentum(lenet.trainable_params(), learning_rate=0.1, momentum=0.9)
186 model = Model(lenet, loss_fn=loss, optimizer=optim, metrics={'acc': Accuracy()})
|
| /third_party/mindspore/tests/ut/python/train/ |
| D | test_training.py | 24 from mindspore.nn.optim import Momentum
75 optim = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
76 model = Model(net, loss_fn=loss, optimizer=optim, metrics=metrics)
|
| /third_party/mindspore/tests/mindspore_test_framework/apps/ |
| D | test_lamb_check_loss.py | 19 from mindspore.nn.optim import Lamb
|
| /third_party/quickjs/ |
| D | TODO | 58 - peephole optim: push_atom_value, to_propkey -> push_atom_value
59 - peephole optim: put_loc x, get_loc_check x -> set_loc x
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pipeline/parse/ |
| D | parse_compile.py | 26 from mindspore.nn.optim import Momentum
|
| /third_party/mindspore/mindspore/nn/probability/toolbox/ |
| D | anomaly_detection.py | 22 from ...optim import Adam
|
| /third_party/mindspore/tests/ut/python/nn/optim/ |
| D | test_momentum.py | 22 from mindspore.nn.optim import Momentum
|
| D | test_lr_schedule.py | 20 from mindspore.nn.optim import Optimizer
|
| D | test_ada_grad.py | 24 from mindspore.nn.optim import Adagrad
|
| D | test_rmsprop.py | 23 from mindspore.nn.optim import RMSProp
|
| /third_party/mindspore/mindspore/boost/ |
| D | grad_freeze.py | 20 from mindspore.nn.optim import Optimizer
23 from mindspore.nn.optim import LARS
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_momentum_op.py | 24 from mindspore.nn.optim import Momentum
|
| D | test_adam_op.py | 24 from mindspore.nn.optim import Adam
|
| D | test_sgd_op.py | 24 from mindspore.nn.optim import SGD
|
| /third_party/ltp/tools/sparse/sparse-src/Documentation/ |
| D | dev-options.rst | 38 * ``optim``
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_sgd_op.py | 24 from mindspore.nn.optim import SGD
|
| D | test_momentum_op.py | 25 from mindspore.nn.optim import Momentum
|