| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_lenet.py | 149 net_with_criterion = WithLossCell(net, criterion)
150 train_network = GradWrap(net_with_criterion)
185 net_with_criterion = WithLossCell(net, criterion)
186 net_with_criterion.set_train()
191 …forward_value_and_grad = nn.ForwardValueAndGrad(network=net_with_criterion, weights=weights, get_b…
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gpu_lenet.py | 141 net_with_criterion = WithLossCell(net, criterion)
142 train_network = TrainOneStepCell(net_with_criterion, optimizer) # optimizer
217 net_with_criterion = WithLossCell(network, criterion)
218 net_with_criterion.set_train()
223 …train_network = ForwardValueAndGrad(network=net_with_criterion, weights=weights, get_by_list=True,…
|
| D | test_network_main.py | 43 net_with_criterion = WithLossCell(net, criterion)
44 train_network = TrainOneStepCell(net_with_criterion, optimizer) # optimizer
|
| D | test_cpu_lenet.py | 64 net_with_criterion = WithLossCell(net, criterion)
65 train_network = TrainOneStepCell(net_with_criterion, optimizer) # optimizer
|
| D | test_gpu_alexnet.py | 80 net_with_criterion = WithLossCell(net, criterion)
81 train_network = TrainOneStepCell(net_with_criterion, optimizer)
|
| D | test_gpu_lstm.py | 133 net_with_criterion = WithLossCell(net, criterion)
134 train_network = TrainOneStepCell(net_with_criterion, optimizer) # optimizer
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_momentum_op.py | 56 net_with_criterion = WithLossCell(net, criterion)
57 train_network = TrainOneStepCell(net_with_criterion, optimizer) # optimizer
|
| D | test_adam_op.py | 53 net_with_criterion = WithLossCell(net, criterion)
55 net_with_criterion, optimizer)
|
| D | test_sgd_op.py | 59 net_with_criterion = WithLossCell(net, criterion)
60 train_network = TrainOneStepCell(net_with_criterion, optimizer) # optimizer
|
| D | test_adam_weight_decay_op.py | 53 net_with_criterion = WithLossCell(net, criterion)
54 train_network = TrainOneStepCell(net_with_criterion, optimizer)
|
| D | test_cpu_type.py | 106 net_with_criterion = WithLossCell(net, criterion)
107 train_network = TrainOneStepCell(net_with_criterion, optimizer) # optimizer
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_sgd_op.py | 59 net_with_criterion = WithLossCell(net, criterion)
60 train_network = TrainOneStepCell(net_with_criterion, optimizer) # optimizer
|
| D | test_momentum_op.py | 56 net_with_criterion = WithLossCell(net, criterion)
57 train_network = TrainOneStepCell(net_with_criterion, optimizer) # optimizer
|
| D | test_ftrl_op.py | 53 net_with_criterion = WithLossCell(net, criterion)
55 net_with_criterion, optimizer)
|
| D | test_adam_op.py | 53 net_with_criterion = WithLossCell(net, criterion)
55 net_with_criterion, optimizer)
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_apply_adam.py | 58 net_with_criterion = WithLossCell(ms_dense, criterion)
59 train_network = TrainOneStepCell(net_with_criterion, optimizer)
|
| /third_party/mindspore/tests/st/ps/cmp_sparse_embedding/ |
| D | test_cmp_sparse_embedding.py | 67 net_with_criterion = WithLossCell(net, criterion)
68 train_network = TrainOneStepCell(net_with_criterion, optimizer)
|
| /third_party/mindspore/tests/st/recompute/ |
| D | test_cpu_lenet_recompute.py | 65 net_with_criterion = WithLossCell(net, criterion)
66 train_network = TrainOneStepCell(net_with_criterion, optimizer) # optimizer
|
| /third_party/mindspore/tests/st/host_device/ |
| D | test_host_device_lenet.py | 70 net_with_criterion = WithLossCell(net, criterion)
71 train_network = TrainOneStepCell(net_with_criterion, optimizer) # optimizer
|
| /third_party/mindspore/tests/st/ps/multi_full_ps/ |
| D | test_multi_full_ps.py | 105 net_with_criterion = WithLossCell(network, criterion) variable
106 train_network = TrainOneStepCell(net_with_criterion, net_opt)
|
| /third_party/mindspore/tests/st/compile_cache/ |
| D | test_ascend_lenet.py | 65 net_with_criterion = WithLossCell(net, criterion)
66 train_network = TrainOneStepCell(net_with_criterion, optimizer) # optimizer
|
| /third_party/mindspore/tests/st/fl/cross_silo_lenet/ |
| D | test_cross_silo_lenet.py | 107 net_with_criterion = WithLossCell(network, criterion) variable
108 train_network = TrainOneStepCell(net_with_criterion, net_opt)
|
| /third_party/mindspore/tests/st/high_grad/ |
| D | test_highgrad_train.py | 111 net_with_criterion = WithLossCellLocal(grad_net, criterion)
112 train_network = TrainOneStepCell(net_with_criterion, optimizer)
|
| /third_party/mindspore/tests/st/nccl/ |
| D | test_nccl_lenet.py | 89 net_with_criterion = WithLossCell(net, criterion)
90 train_network = TrainOneStepCell(net_with_criterion, mom_optimizer)
|
| /third_party/mindspore/tests/st/fl/mobile/ |
| D | test_mobile_lenet.py | 133 net_with_criterion = WithLossCell(network, criterion) variable
134 train_network = TrainOneStepCell(net_with_criterion, net_opt)
|