Home
last modified time | relevance | path

Searched refs:TrainOneStepCell (Results 1 – 25 of 135) sorted by relevance

123456

/third_party/mindspore/tests/ut/python/nn/optim/
Dtest_adafactor.py22 from mindspore.nn import TrainOneStepCell, WithLossCell
84 train_network = TrainOneStepCell(net_with_loss, optimizer)
99 train_network = TrainOneStepCell(net_with_loss, optimizer)
115 train_network = TrainOneStepCell(net_with_loss, optimizer)
135 train_network = TrainOneStepCell(net_with_loss, optimizer)
155 train_network = TrainOneStepCell(net_with_loss, optimizer)
175 train_network = TrainOneStepCell(net_with_loss, optimizer)
194 train_network = TrainOneStepCell(net_with_loss, optimizer)
212 train_network = TrainOneStepCell(net_with_loss, optimizer)
229 train_network = TrainOneStepCell(net_with_loss, optimizer)
[all …]
Dtest_adam.py22 from mindspore.nn import TrainOneStepCell, WithLossCell
88 train_network = TrainOneStepCell(net_with_loss, optimizer)
103 train_network = TrainOneStepCell(net_with_loss, optimizer)
116 train_network = TrainOneStepCell(net, optimizer)
128 train_network = TrainOneStepCell(net, optimizer)
148 train_network = TrainOneStepCell(net_with_loss, optimizer)
166 train_network = TrainOneStepCell(net_with_loss, optimizer)
184 train_network = TrainOneStepCell(net_with_loss, optimizer)
202 train_network = TrainOneStepCell(net_with_loss, optimizer)
Dtest_proximal_ada_grad.py22 from mindspore.nn import TrainOneStepCell, WithLossCell
68 train_network = TrainOneStepCell(net_with_loss, optimizer)
81 train_network = TrainOneStepCell(net, optimizer)
93 train_network = TrainOneStepCell(net, optimizer)
Dtest_ftrl.py22 from mindspore.nn import TrainOneStepCell, WithLossCell
68 train_network = TrainOneStepCell(net_with_loss, optimizer)
81 train_network = TrainOneStepCell(net, optimizer)
94 train_network = TrainOneStepCell(net, optimizer)
Dtest_lamb.py21 from mindspore.nn import TrainOneStepCell, WithLossCell
85 train_network = TrainOneStepCell(net_with_loss, optimizer)
100 train_network = TrainOneStepCell(net_with_loss, optimizer)
118 train_network = TrainOneStepCell(net_with_loss, optimizer)
Dtest_lars.py22 from mindspore.nn import TrainOneStepCell, WithLossCell
63 train_network = TrainOneStepCell(net_with_loss, optimizer)
80 train_network = TrainOneStepCell(net_with_loss, optimizer)
Dtest_momentum.py21 from mindspore.nn import TrainOneStepCell, WithLossCell
52 train_network = TrainOneStepCell(net_with_loss, optimizer)
/third_party/mindspore/tests/ut/python/communication/
Dtest_comm.py27 from mindspore.nn import TrainOneStepCell, WithLossCell
156 network = TrainOneStepCell(network, optimizer)
180 network = TrainOneStepCell(network, optimizer)
194 network = TrainOneStepCell(network, optimizer)
209 network = TrainOneStepCell(network, optimizer)
230 network = TrainOneStepCell(network, optimizer)
245 network = TrainOneStepCell(network, optimizer)
/third_party/mindspore/tests/st/dynamic_shape/
Dtest_ftrl.py19 from mindspore.nn import TrainOneStepCell
49 train_network = TrainOneStepCell(net, optimizer)
70 train_network = TrainOneStepCell(net, optimizer)
89 train_network = TrainOneStepCell(net, optimizer)
Dtest_dynamic_shape_embedding.py19 from mindspore.nn import TrainOneStepCell, WithLossCell
52 train_network1 = TrainOneStepCell(WithLossCell(net1, loss), optimizer1)
60 train_network2 = TrainOneStepCell(WithLossCell(net2, loss), optimizer2)
/third_party/mindspore/tests/ut/python/parallel/
Dtest_loss_and_optimizer.py22 from mindspore.nn import TrainOneStepCell
73 train_net = TrainOneStepCell(net_with_loss, optimizer)
107 train_net = TrainOneStepCell(net_with_loss, optimizer)
142 train_net = TrainOneStepCell(net_with_loss, optimizer)
178 train_net = TrainOneStepCell(net_with_loss, optimizer)
213 train_net = TrainOneStepCell(net_with_loss, optimizer)
Dtest_parallel_optimizer.py22 from mindspore.nn import TrainOneStepCell, WithLossCell
92 train_network = TrainOneStepCell(net, optimizer).set_comm_fusion(4)
156 train_network = TrainOneStepCell(net_with_loss, optimizer)
172 train_network = TrainOneStepCell(net_with_loss, optimizer)
189 train_network = TrainOneStepCell(net_with_loss, optimizer)
Dtest_broadcast_to.py21 from mindspore.nn import TrainOneStepCell, Momentum
69 train_net = TrainOneStepCell(net, optimizer)
79 train_net = TrainOneStepCell(net, optimizer)
Dtest_optimizer_clone_weight.py22 from mindspore.nn import TrainOneStepCell
73 train_net = TrainOneStepCell(net_with_loss, optimizer)
108 train_net = TrainOneStepCell(net_with_loss, optimizer)
Dtest_auto_parallel_inference.py20 from mindspore.nn import WithLossCell, TrainOneStepCell
49 train_network = TrainOneStepCell(net_with_loss, optimizer)
Dtest_reshape_optimized.py20 from mindspore.nn import Cell, TrainOneStepCell, Momentum
44 train_net = TrainOneStepCell(net, optimizer)
Dtest_parallel_moe.py23 from mindspore.nn.wrap.cell_wrapper import TrainOneStepCell, _VirtualDatasetCell
91 net_with_grad = TrainOneStepCell(net, optimizer=optimizer)
123 net_with_grad = TrainOneStepCell(net, optimizer=optimizer)
Dtest_batchnorm.py20 from mindspore.nn import Cell, TrainOneStepCell, Momentum, BatchNorm2d, BatchNorm1d
47 train_net = TrainOneStepCell(net, optimizer)
97 train_net = TrainOneStepCell(net, optimizer)
Dtest_auto_parallel_activation.py20 from mindspore.nn import Cell, TrainOneStepCell, Momentum
44 train_net = TrainOneStepCell(net, optimizer)
/third_party/mindspore/tests/st/pynative/dynamic_shape/
Dtest_pynative_ftrl.py19 from mindspore.nn import TrainOneStepCell
49 train_network = TrainOneStepCell(net, optimizer)
70 train_network = TrainOneStepCell(net, optimizer)
/third_party/mindspore/tests/ut/python/nn/
Dtest_cell_wrapper.py22 from mindspore.nn import TrainOneStepCell, WithLossCell, ParameterUpdate
47 train_network = TrainOneStepCell(net_with_loss, optimizer)
78 train_network = TrainOneStepCell(net_with_loss, optimizer)
/third_party/mindspore/tests/st/debugger/golden/
Dasync_sink_mode_true_read_tensors.expected3 node name = Default/network-TrainOneStepCell/network-WithLossCell/_backbone-AlexNet/conv3-Conv2d/C…
17 node name = Default/network-TrainOneStepCell/network-WithLossCell/_backbone-AlexNet/ReLUV2-op348
/third_party/mindspore/mindspore/nn/probability/infer/variational/
Dsvi.py20 from ....wrap.cell_wrapper import TrainOneStepCell
64 train_net = TrainOneStepCell(self.net_with_loss, self.optimizer)
/third_party/mindspore/tests/st/fl/cross_silo_lenet/src/
Dcell_wrapper.py18 from mindspore.nn import TrainOneStepCell
20 class TrainOneStepCellForFLWorker(TrainOneStepCell):
/third_party/mindspore/tests/ut/python/utils/
Dtest_export.py71 class TrainOneStepCell(nn.Cell): class
73 super(TrainOneStepCell, self).__init__(auto_prefix=False)
93 net = TrainOneStepCell(WithLossCell(network))

123456