Home
last modified time | relevance | path

Searched refs:get_parameters (Results 1 – 25 of 97) sorted by relevance

1234

/third_party/mindspore/tests/ut/python/communication/
Dtest_comm.py152 optimizer = Momentum(filter(lambda x: x.requires_grad, network.get_parameters()),
176 optimizer = Momentum(filter(lambda x: x.requires_grad, network.get_parameters()),
190 optimizer = Momentum(filter(lambda x: x.requires_grad, network.get_parameters()),
205 optimizer = Momentum(filter(lambda x: x.requires_grad, network.get_parameters()),
226 optimizer = Momentum(filter(lambda x: x.requires_grad, network.get_parameters()),
241 optimizer = Momentum(filter(lambda x: x.requires_grad, network.get_parameters()),
/third_party/mindspore/tests/ut/python/nn/optim/
Dtest_rmsprop.py60 RMSProp(net.get_parameters(), momentum=-0.1, learning_rate=0.1, weight_decay=0.9)
63 RMSProp(net.get_parameters(), momentum=1, learning_rate=0.1, weight_decay=0.9)
Dtest_adam.py209 AdamWeightDecay(net.get_parameters(), beta1=1.0, learning_rate=0.1)
215 AdamWeightDecay(net.get_parameters(), beta2=1.0, learning_rate=0.1)
221 AdamWeightDecay(net.get_parameters(), eps=-0.1, learning_rate=0.1)
227 AdamWeightDecay(net.get_parameters())
Dtest_lr_schedule.py57 self.weights = ParameterTuple(network.get_parameters())
/third_party/mindspore/tests/st/pynative/
Dtest_pynative_temporary_cell.py73 optimizer = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.1, 0.9)
80 …compare_optimizer = Momentum(filter(lambda x: x.requires_grad, compare_net.get_parameters()), 0.1,…
Dtest_pynative_lenet.py124 self.weights = ParameterTuple(filter(lambda x: x.requires_grad, network.get_parameters()))
147 optimizer = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.1, 0.9)
188 weights = ParameterTuple(filter(lambda x: x.requires_grad, net.get_parameters()))
Dtest_pynative_hook.py111 self.weights = ParameterTuple(filter(lambda x: x.requires_grad, network.get_parameters()))
173 optimizer = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.1, 0.9)
/third_party/mindspore/tests/ut/python/nn/
Dtest_cell_wrapper.py44 …optimizer = Momentum(net.get_parameters(), Tensor(np.array([0.1, 0.01, 0.001]), mstype.float32), 0…
75 optimizer = Momentum(net.get_parameters(), 0.01, 0.001)
/third_party/mindspore/tests/ut/python/parallel/
Dtest_auto_parallel_resnet.py302 opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.01, 0.9)
678 opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.01, 0.9)
706 opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.01, 0.9)
732 opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.01, 0.9)
758 opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.01, 0.9)
Dtest_loss_scale.py187 opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.01, 0.9)
200 opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.01, 0.9)
/third_party/mindspore/tests/st/pynative/ms_function/
Dtest_pynative_ms_function.py83 optimizer = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.1, 0.9)
140 optimizer = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.1, 0.9)
195 optimizer = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.1, 0.9)
Dtest_pynative_lenet_ms_function.py157 self.weights = ParameterTuple(filter(lambda x: x.requires_grad, network.get_parameters()))
180 optimizer = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.1, 0.9)
/third_party/mindspore/tests/st/nontask_sink/
Dtest_lenet.py123 self.weights = ParameterTuple(filter(lambda x: x.requires_grad, network.get_parameters()))
138 optimizer = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.1, 0.9)
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pipeline/parse/
Dparse_compile.py52 opt = Momentum(net.get_parameters(), learning_rate=0.1, momentum=0.9)
/third_party/mindspore/tests/ut/python/pynative_mode/
Dtest_hook.py134 self.weights = ParameterTuple(filter(lambda x: x.requires_grad, network.get_parameters()))
143 optimizer = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), 0.1, 0.9)
Dtest_ms_function.py62 … self.optimizer = Momentum(filter(lambda x: x.requires_grad, self.net.get_parameters()), 0.1, 0.9)
/third_party/mindspore/tests/st/quantization/ops/
Dtest_Conv2dBnFoldQuant.py41 …opt = nn.Momentum(filter(lambda x: x.requires_grad, network.get_parameters()), learning_rate=0.1, …
/third_party/mindspore/tests/st/ops/cpu/
Dtest_momentum_op.py54 …optimizer = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), learning_rate, momen…
Dtest_adam_op.py51 net.get_parameters()), learning_rate=0.01)
Dtest_adam_weight_decay_op.py51 …optimizer = AdamWeightDecayOp(filter(lambda x: x.requires_grad, net.get_parameters()), learning_ra…
Dtest_sgd_op.py56 …optimizer = SGD(filter(lambda x: x.requires_grad, net.get_parameters()), learning_rate, momentum, …
/third_party/mindspore/tests/st/networks/models/resnet50/src_thor/
Dthor.py161 params = filter(lambda x: x.requires_grad, net.get_parameters())
171 … self.matrix_A_cov = ParameterTuple(filter(lambda x: 'matrix_a' in x.name, net.get_parameters()))
172 … self.matrix_G_cov = ParameterTuple(filter(lambda x: 'matrix_g' in x.name, net.get_parameters()))
173 …elf.A_normalizer = ParameterTuple(filter(lambda x: 'a_normalizer' in x.name, net.get_parameters()))
174 …elf.G_normalizer = ParameterTuple(filter(lambda x: 'g_normalizer' in x.name, net.get_parameters()))
/third_party/mindspore/tests/st/ps/part_ps/
Dtest_ps_embedding_heterogeneous_conv2d_adam.py147 opt = Adam(params=filter(lambda x: x.requires_grad, net.get_parameters()))
164 opt = Adam(params=filter(lambda x: x.requires_grad, net.get_parameters()))
/third_party/mindspore/tests/st/ops/gpu/
Dtest_momentum_op.py54 …optimizer = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), learning_rate, momen…
Dtest_sgd_op.py56 …optimizer = SGD(filter(lambda x: x.requires_grad, net.get_parameters()), learning_rate, momentum, …

1234