| /third_party/mindspore/tests/ut/python/exec/ |
| D | test_train_with_lars.py | 74 def lr_gen(fn, epoch_size): argument
79 def me_train_tensor(net, input_np, label_np, epoch_size=2): argument
|
| D | test_train.py | 27 def lr_gen(fn, epoch_size): argument
32 def me_train_tensor(net, input_np, label_np, epoch_size=2): argument
|
| /third_party/mindspore/tests/st/networks/models/bert/src/ |
| D | dataset.py | 26 def create_bert_dataset(epoch_size=1, device_num=1, rank=0, do_shuffle="true", data_dir=None, schem… argument
|
| /third_party/mindspore/tests/st/networks/models/resnet50/ |
| D | test_resnet50_imagenet.py | 132 def train_process(q, device_id, epoch_size, device_num, enable_hccl): argument
230 def train_process_thor(q, device_id, epoch_size, device_num, enable_hccl): argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/deeplabv3/ |
| D | train_one_epoch_with_loss.py | 73 epoch_size = 3 variable
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | test_resnet_cifar_1p.py | 134 def train_process(epoch_size, num_classes, batch_size): argument
|
| D | resnet_cifar.py | 131 epoch_size = args_opt.epoch_size variable
|
| D | test_resnet_cifar_8p.py | 145 def train_process(q, device_id, epoch_size, num_classes, device_num, batch_size, enable_hccl): argument
|
| /third_party/mindspore/tests/st/mem_reuse/ |
| D | resnet_cifar_memreuse.py | 130 epoch_size = args_opt.epoch_size variable
|
| D | resnet_cifar_normal.py | 130 epoch_size = args_opt.epoch_size variable
|
| /third_party/mindspore/tests/st/fusion/ |
| D | test_conv_bn1_fusion.py | 41 def me_train_tensor(net, input_np, label_np, epoch_size=2): argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | lr_scheduler.py | 58 def step_lr(lr, epoch_size, steps_per_epoch, max_epoch, gamma=0.1): argument
|
| /third_party/mindspore/tests/st/networks/models/bert/bert_performance/ |
| D | test_bert_thor.py | 143 def train_process_bert_thor(q, device_id, epoch_size, device_num): argument
|
| /third_party/mindspore/tests/st/ps/part_ps/ |
| D | test_ps_embedding_heterogeneous_conv2d_adam.py | 127 epoch_size=1, target='CPU', sparse=True): argument
|