MindSpore训练报错RuntimeError: For 'Gather', the 'input_indices' should be in the range [0, 16), but got 16, error_code[58]
收藏回复举报
MindSpore训练报错RuntimeError: For 'Gather', the 'input_indices' should be in the range [0, 16), but got 16, error_code[58]
发表于2024-09-05 17:17:07
0 查看

1.  系统环境 

硬件环境(Ascend/GPU/CPU): Ascend/GPU/CPU

MindSpore版本:2.2.10

执行模式(PyNative/ Graph):不限

Python版本:不限

操作系统平台:Linux 

2.  报错信息 

2.1 问题描述及报错

参考https://github.com/mindspore-lab/mindyolo。根据适配选择的mindspore=2.2.10,mindyolo=0.3.0。但是出现了如下报错

Traceback (most recent call last):
    File "C: \Users\24305\PycharmProjects\yolov5_6\train\train.py", line 314, in <module>
        train(args)
    File "C: \Users\24305\PycharmProjects\yolov5_6\train\train.py", line 269, in train
        trainer.train(
    File "C: \Users\24305\.anaconda\envslmindspore_py39_2.2.10\lib\site-packages\mindyolo\utils\trainer_factory.py", line 170, in train
        run_context.loss, run_context.ln = self.train_step(imgs, labels, segments,
    File "C: \Users\24305\.anacond a\envs\mindspore_py39_2.2.10\lib\site-packages\mindyolo\utils\trainer_factory.py", line 427, in train_step
        cur_lr = lr_cell(Tensor(self .global_step, ms.int32)) -asnumpy().item()
    File "C:\Users\24305\ .anaconda\envs\mindspore_py3 9_2.2.10\lib\site-packages\mindspore\nn\cell.py", line 680, in -_call_-
        out = self.compile_and_run (*args, **kwargs)
    File "c: \Users\24305\.anacondalenvslmindspore_py39_2.2.10\lib\site-packages\mindspore\nn\cell.py", line 1023, in compile_and_run
        return cell_graph_executor (self, *new_args, phase=self.phase)
    File "c: \Users\24305\.anaconda\envs\mindspore_py39_2.2.10\lib\site-packages\mindspore\common\api.py", line 1589, in --call_-
        return self.run(obj, *args, phase=phase)
    File "C: \Users\24305\.anaconda\envs\mindspore_py39_2.2.10\lib\site-packages\mindspore\commonlapi.py", line 1628, in run
        return self._exec_pip(obj, *args, phase=phase_real)
    File "c: \Users\24305\.anaconda\envs\mindspore_py39_2.2.10\lib\site-packages\mindspore\commonlapi.py", line 121, in wrapper
        results = fn(*arg, **kwargs)
    File "C: \Users\24305\.anaconda\envs\mindspore_py39_2.2.10\lib\site-packages\mindspore\common\api.py", line 1628, in run
        return self._exec_pip(obj, *args, phase=phase_real)
    File "C: \Users\24305\.anaconda\envs\mindspore_py39_2.2.10\lib\site-packages\mindspore\common\api.py", line 121, in wrapper
        results = fn(*arg, **kwargs)
    File "C: \Users\24305\.anacondalenvs\mindspore_py39. 2.2.10\lib\site-packages\mindsporelcommonlapi.py", line 1608, in _exec_pip
        return self._graph_executor (args, phase)
RuntimeError: For 'Gather' the 'input indices' should be in the range [o. 16) but got 16. error code[58]

根因分析

******此处由用户填写******

解决方案

******此处由用户填写******

包含文字方案和最终脚本代码

请将正确的脚本打包并上传附件

本帖最后由 匿名用户2024/09/06 09:29:39 编辑

我要发帖子