| /third_party/mindspore/mindspore/nn/optim/ |
| D | sgd.py | 139 …def __init__(self, params, learning_rate=0.1, momentum=0.0, dampening=0.0, weight_decay=0.0, neste… argument
152 if isinstance(dampening, int):
153 dampening = float(dampening)
154 if not isinstance(dampening, float):
157 if dampening < 0.0:
158 … raise ValueError("dampening should be at least 0.0, but got dampening {}".format(dampening))
159 self.dampening = dampening
166 if nesterov and (momentum <= 0.0 or dampening != 0.0):
168 "but got momentum {}, dampening {}".format(momentum, dampening))
171 self.opt = P.SGD(dampening, weight_decay, nesterov)
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | sgd.cc | 21 void SGD::Init(const float dampening, const float weight_decay, const bool nesterov) { in Init() argument
23 set_dampening(dampening); in Init()
27 void SGD::set_dampening(const float dampening) { in set_dampening() argument
28 …if (get_nesterov()) CheckAndConvertUtils::CheckValue<float>(kDampening, dampening, kEqual, 0.0, na… in set_dampening()
29 (void)AddAttr(kDampening, MakeValue(dampening)); in set_dampening()
|
| D | sgd.h | 38 …void Init(const float dampening = 0.0, const float weight_decay = 0.0, const bool nesterov = false…
40 void set_dampening(const float dampening);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | sgd_impl.cu | 21 __global__ void SGDKernel(const int size, const T dampening, const T weight_decay, const bool neste… in SGDKernel() argument
34 accum[i] = accum[i] * momentum[0] + (1.0 - dampening) * grad_new; in SGDKernel()
49 void SGD(const int size, const T dampening, const T weight_decay, const bool nesterov, const T *lr,… in SGD() argument
51 …SGDKernel<<<GET_BLOCKS(size), GET_THREADS, 0, cuda_stream>>>(size, dampening, weight_decay, nester… in SGD()
55 template void SGD(const int size, const float dampening, const float weight_decay, const bool neste…
|
| D | sgd_impl.cuh | 23 void SGD(const int size, const T dampening, const T weight_decay, const bool nesterov, const T *lr,…
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_sgd_op.py | 51 dampening = 0.0
56 … = SGD(filter(lambda x: x.requires_grad, net.get_parameters()), learning_rate, momentum, dampening,
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_sgd_op.py | 51 dampening = 0.0
56 … = SGD(filter(lambda x: x.requires_grad, net.get_parameters()), learning_rate, momentum, dampening,
|
| /third_party/mindspore/tests/ut/python/nn/optim/ |
| D | test_optimizer.py | 67 … SGD(params, learning_rate=0.1, momentum=-0.1, dampening=0, weight_decay=0, nesterov=False)
69 … SGD(params, learning_rate=0.12, momentum=-0.1, dampening=0, weight_decay=0, nesterov=False)
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | sgd.cc | 33 …ight, float *accumulate, const float *gradient, float learning_rate, float dampening, float moment, in DoSgd() argument
38 accumulate[i] = accumulate[i] * moment + gradient[i] * (1.f - dampening); in DoSgd()
43 accumulate[i] = accumulate[i] * moment + gradient[i] * (1.f - dampening); in DoSgd()
|
| /third_party/mindspore/mindspore/lite/examples/export_models/models/ |
| D | effnet_train_export.py | 29 optimizer = nn.SGD(n.trainable_params(), learning_rate=0.01, momentum=0.9, dampening=0.0, weight_de…
|
| D | nin_train_export.py | 29 optimizer = nn.SGD(n.trainable_params(), learning_rate=0.01, momentum=0.9, dampening=0.0, weight_de…
|
| D | xception_train_export.py | 32 optimizer = nn.SGD(n.trainable_params(), learning_rate=0.01, momentum=0.9, dampening=0.0, weight_de…
|
| D | densenet_train_export.py | 32 optimizer = nn.SGD(n.trainable_params(), learning_rate=0.001, momentum=0.9, dampening=0.0, weight_d…
|
| D | googlenet_train_export.py | 29 optimizer = nn.SGD(n.trainable_params(), learning_rate=0.01, momentum=0.9, dampening=0.0, weight_de…
|
| D | mobilenetv1_train_export.py | 29 optimizer = nn.SGD(n.trainable_params(), learning_rate=1e-2, momentum=0.9, dampening=0.1, weight_de…
|
| D | resnet_train_export.py | 30 optimizer = nn.SGD(n.trainable_params(), learning_rate=0.01, momentum=0.9, dampening=0.0, weight_de…
|
| D | effnet_tune_train_export.py | 60 dampening=0.01, weight_decay=0.0, nesterov=False, loss_scale=1.0)
|
| /third_party/mindspore/mindspore/lite/examples/transfer_learning/model/ |
| D | transfer_learning_export.py | 53 dampening=0.01, weight_decay=0.0, nesterov=False, loss_scale=1.0)
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | nn_training_ops_declare.cc | 144 ATTR_MAP(SGD) = {{"dampening", ATTR_DESC(dampening, AnyTraits<float>())},
|
| /third_party/mindspore/mindspore/lite/src/train/ |
| D | train_populate_parameter.cc | 140 p->dampening_ = value->dampening(); in PopulateSgdParameter()
|
| D | train_populate_parameter_v0.cc | 219 p->dampening_ = sgd_prim->dampening(); in PopulateSgdParameter()
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | ops.fbs | 882 dampening: float;
|
| D | ops_v0.fbs | 968 dampening: float;
|
| D | ops_v0_generated.h | 9895 float dampening() const { in dampening() function
9917 void add_dampening(float dampening) { in add_dampening()
9918 fbb_.AddElement<float>(Sgd::VT_DAMPENING, dampening, 0.0f); in add_dampening()
9937 float dampening = 0.0f,
9940 builder_.add_dampening(dampening);
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_def.cc | 882 OP_ATTR(dampening, float)
|