报错内容
RuntimeError: For 'GetNext', get data timeout. Queue name: 9d263f4a-46dc-11ed-b99c-0242ac110002 ---------------------------------------------------- - C++ Call Stack: (For framework developers) ---------------------------------------------------- mindspore/ccsrc/plugin/device/gpu/kernel/data/dataset_iterator_kernel.cc:135 ReadDevice
看不出问题代码是哪部分,因为是dataset_iterator_kernel的问题,请问出错的地方是数据加载吗
用官网的例子修改的函数如下
def create_dataset(dataset_path, spilt, repeat_num=1, batch_size=32,num_classes=6):
"""
Create a train or eval dataset.
Args:
dataset_path (str): The path of dataset.
spilt (str): Whether dataset is used for train or eval.
repeat_num (int): The repeat times of dataset. Default: 1.
batch_size (int): The batch size of dataset. Default: 32.
Returns:
Datase