Searched refs:create_bert_dataset (Results 1 – 2 of 2) sorted by relevance
| /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… function
|
| /third_party/mindspore/tests/st/networks/models/bert/bert_performance/ |
| D | test_bert_thor.py | 108 def create_bert_dataset(device_num=1, rank=0, do_shuffle="true", data_dir=None, schema_dir=None): function
160 …data_set = create_bert_dataset(device_num=device_num, rank=rank, do_shuffle=False, data_dir=DATASE…
|