自定义数据集训练时出现错误

我在使用自定义的数据集进行训练的时候出现了以下错误：
```
The function call stack (See file '/root/mindyolo/rank_0/om/analyze_fail.ir' for more details. Get instructions about `analyze_fail.ir` at https://www.mindspore.cn/search?inputValue=analyze_fail.ir):
# 0 In file /root/mindyolo/mindyolo/utils/train_step_factory.py:77
            return train_step_func(*args)
                   ^
# 1 In file /root/mindyolo/mindyolo/utils/train_step_factory.py:59
            if clip_grad:
# 2 In file /root/mindyolo/mindyolo/utils/train_step_factory.py:54
            (loss, loss_items), grads = grad_fn(x, label)
                                        ^
# 3 In file /root/miniconda3/lib/python3.8/site-packages/mindspore/ops/composite/base.py:584
                        return grad_(fn, weights)(*args)
                               ^
# 4 In file /root/mindyolo/mindyolo/utils/train_step_factory.py:47
            loss, loss_items = loss_fn(pred, label, x)
                               ^
# 5 In file /root/mindyolo/mindyolo/models/losses/yolov5_loss.py:88
        for layer_index, pi in enumerate(p):  # layer index, layer predictions
        ^
# 6 In file /root/mindyolo/mindyolo/utils/train_step_factory.py:47
            loss, loss_items = loss_fn(pred, label, x)
                               ^
# 7 In file /root/mindyolo/mindyolo/models/losses/yolov5_loss.py:88
        for layer_index, pi in enumerate(p):  # layer index, layer predictions
        ^
# 8 In file /root/mindyolo/mindyolo/utils/train_step_factory.py:47
            loss, loss_items = loss_fn(pred, label, x)
                               ^
# 9 In file /root/mindyolo/mindyolo/models/losses/yolov5_loss.py:88
        for layer_index, pi in enumerate(p):  # layer index, layer predictions
        ^
# 10 In file /root/mindyolo/mindyolo/utils/train_step_factory.py:47
            loss, loss_items = loss_fn(pred, label, x)
                               ^
# 11 In file /root/mindyolo/mindyolo/models/losses/yolov5_loss.py:88
        for layer_index, pi in enumerate(p):  # layer index, layer predictions
        ^
# 12 In file /root/mindyolo/mindyolo/models/losses/yolov5_loss.py:133
        loss_item = ops.stop_gradient(ops.stack((loss, lbox, lobj, lcls)))
                                      ^
# 13 In file /root/miniconda3/lib/python3.8/site-packages/mindspore/ops/function/array_func.py:2207
    return _stack(tensors)
           ^

Traceback (most recent call last):
  File "/root/mindyolo/train.py", line 321, in <module>
    train(args)
  File "/root/mindyolo/train.py", line 276, in train
    trainer.train(
  File "/root/mindyolo/mindyolo/utils/trainer_factory.py", line 170, in train
    run_context.loss, run_context.lr = self.train_step(imgs, labels, segments,
  File "/root/mindyolo/mindyolo/utils/trainer_factory.py", line 366, in train_step
    loss, loss_item, _, grads_finite = self.train_step_fn(imgs, labels, True)
  File "/root/miniconda3/lib/python3.8/site-packages/mindspore/common/api.py", line 807, in staging_specialize
    out = _MindsporeFunctionExecutor(func, hash_obj, input_signature, process_obj, jit_config)(*args, **kwargs)
  File "/root/miniconda3/lib/python3.8/site-packages/mindspore/common/api.py", line 106, in wrapper
    results = fn(*arg, **kwargs)
  File "/root/miniconda3/lib/python3.8/site-packages/mindspore/common/api.py", line 526, in __call__
    raise err
  File "/root/miniconda3/lib/python3.8/site-packages/mindspore/common/api.py", line 523, in __call__
    phase = self.compile(self.fn.__name__, *args_list, **kwargs)
  File "/root/miniconda3/lib/python3.8/site-packages/mindspore/common/api.py", line 599, in compile
    is_compile = self._graph_executor.compile(self.fn, compile_args, kwargs, phase, True)
  File "/root/miniconda3/lib/python3.8/site-packages/mindspore/ops/operations/array_ops.py", line 3108, in __infer__
    all_shape = _get_stack_shape(value, x_shape, x_type, self.axis, self.name)
  File "/root/miniconda3/lib/python3.8/site-packages/mindspore/ops/operations/array_ops.py", line 3002, in _get_stack_shape
    raise TypeError("For \'{}\', all types should be same, but got {}".format(prim_name, x_type))
TypeError: For 'Stack', all types should be same, but got (mindspore.tensor[float32], mindspore.tensor[float32], mindspore.tensor[float32], mindspore.float32)
Error in atexit._run_exitfuncs:
Traceback (most recent call last):
  File "/root/miniconda3/lib/python3.8/multiprocessing/util.py", line 357, in _exit_function
    p.join()
  File "/root/miniconda3/lib/python3.8/multiprocessing/process.py", line 146, in join
    self._check_closed()
  File "/root/miniconda3/lib/python3.8/multiprocessing/process.py", line 101, in _check_closed
    raise ValueError("process object is closed")
ValueError: process object is closed
```

MindSpore Lab/mindyolo

内容风险标识

评论 (0)