训练到第23个epoch中止,无法正常运行

***************************************************解答*****************************************************看不出问题代码是哪部分,因为是dataset_iterator_kernel的问题,请问出错的地方是数据加载吗。单独迭代一下数据集看看耗时呢?用官网的例子修改的函数如下。看一下是否输出数据时间太长。
摘要由CSDN通过智能技术生成

报错内容

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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值