[Bug]: yolov8s训练异常

### 问题描述

使用master分支代码及modelarts进行训练，出现如下问题
1.采用ms2.0.1-CANN6.0的镜像时，读取数据阶段报错；
2.采用ms2.1.1-CANN6.4的镜像时，构建模型阶段报错。

### 环境信息

- **Hardware Environment(`Ascend`/`GPU`/`CPU`)  / 硬件环境**: 
> /device Ascend 910

- **Software Environment / 软件环境 (Mandatory / 必填)**:
-- MindSpore 2.0.1和2.1.1 :
-- Python version 3.9 :
-- OS platform and distribution Linux notebook-aeaf4f7e-a574-4c04-827a-c9d72a8a7b95 4.19.36-vhulk1907.1.0.h619.eulerosv2r8.aarch64 SMP Mon Jul 22 00:00:00 UTC 2019 aarch64 aarch64 aarch64 GNU/Linux:

- **Execute Mode / 执行模式 (Mandatory / 必填)(`PyNative`/`Graph`)**: 
> /mode graph

### 关联用例

无

### 重现步骤

1.准备yolo格式数据和配置文件
2.开展训练，启动命令为python train.py --config ./configs/yolov8/yolov8s.yaml

### 预期结果

训练正常启动

### 日志/截图

ms版本为2.0.1时，日志为：
2024-01-31 16:17:11,298 [INFO] parse_args:
2024-01-31 16:17:11,298 [INFO] task                                    detect
2024-01-31 16:17:11,298 [INFO] device_target                           Ascend
2024-01-31 16:17:11,298 [INFO] save_dir                                ./runs/2024.01.31-16.17.11
2024-01-31 16:17:11,298 [INFO] device_per_servers                      8
2024-01-31 16:17:11,298 [INFO] log_level                               INFO
2024-01-31 16:17:11,298 [INFO] is_parallel                             False
2024-01-31 16:17:11,298 [INFO] ms_mode                                 0
2024-01-31 16:17:11,298 [INFO] ms_amp_level                            O0
2024-01-31 16:17:11,298 [INFO] keep_loss_fp32                          True
2024-01-31 16:17:11,298 [INFO] ms_loss_scaler                          static
2024-01-31 16:17:11,298 [INFO] ms_loss_scaler_value                    1024.0
2024-01-31 16:17:11,298 [INFO] ms_jit                                  True
2024-01-31 16:17:11,298 [INFO] ms_enable_graph_kernel                  False
2024-01-31 16:17:11,298 [INFO] ms_datasink                             False
2024-01-31 16:17:11,298 [INFO] overflow_still_update                   True
2024-01-31 16:17:11,298 [INFO] clip_grad                               False
2024-01-31 16:17:11,298 [INFO] clip_grad_value                         10.0
2024-01-31 16:17:11,298 [INFO] ema                                     True
2024-01-31 16:17:11,298 [INFO] weight                                  
2024-01-31 16:17:11,298 [INFO] ema_weight                              
2024-01-31 16:17:11,298 [INFO] freeze                                  []
2024-01-31 16:17:11,298 [INFO] epochs                                  500
2024-01-31 16:17:11,298 [INFO] per_batch_size                          16
2024-01-31 16:17:11,298 [INFO] img_size                                640
2024-01-31 16:17:11,298 [INFO] nbs                                     64
2024-01-31 16:17:11,298 [INFO] accumulate                              1
2024-01-31 16:17:11,298 [INFO] auto_accumulate                         False
2024-01-31 16:17:11,298 [INFO] log_interval                            100
2024-01-31 16:17:11,298 [INFO] single_cls                              False
2024-01-31 16:17:11,298 [INFO] sync_bn                                 False
2024-01-31 16:17:11,298 [INFO] keep_checkpoint_max                     100
2024-01-31 16:17:11,298 [INFO] run_eval                                False
2024-01-31 16:17:11,298 [INFO] conf_thres                              0.001
2024-01-31 16:17:11,298 [INFO] iou_thres                               0.7
2024-01-31 16:17:11,298 [INFO] conf_free                               True
2024-01-31 16:17:11,298 [INFO] rect                                    False
2024-01-31 16:17:11,298 [INFO] nms_time_limit                          20.0
2024-01-31 16:17:11,298 [INFO] recompute                               False
2024-01-31 16:17:11,298 [INFO] recompute_layers                        0
2024-01-31 16:17:11,298 [INFO] seed                                    2
2024-01-31 16:17:11,298 [INFO] summary                                 True
2024-01-31 16:17:11,298 [INFO] profiler                                False
2024-01-31 16:17:11,298 [INFO] profiler_step_num                       1
2024-01-31 16:17:11,298 [INFO] opencv_threads_num                      0
2024-01-31 16:17:11,298 [INFO] strict_load                             True
2024-01-31 16:17:11,298 [INFO] enable_modelarts                        False
2024-01-31 16:17:11,298 [INFO] data_url                                
2024-01-31 16:17:11,298 [INFO] ckpt_url                                
2024-01-31 16:17:11,298 [INFO] multi_data_url                          
2024-01-31 16:17:11,298 [INFO] pretrain_url                            
2024-01-31 16:17:11,298 [INFO] train_url                               
2024-01-31 16:17:11,298 [INFO] data_dir                                /cache/data/
2024-01-31 16:17:11,298 [INFO] ckpt_dir                                /cache/pretrain_ckpt/
2024-01-31 16:17:11,298 [INFO] data.dataset_name                       increment
2024-01-31 16:17:11,298 [INFO] data.train_set                          /home/ma-user/work/ljy/mindyolo-master/mindyolo/dataset/increment/increment_V1_total/train2017.txt
2024-01-31 16:17:11,298 [INFO] data.val_set                            /home/ma-user/work/ljy/mindyolo-master/mindyolo/dataset/increment/increment_V1_total/val2017.txt
2024-01-31 16:17:11,298 [INFO] data.test_set                           ./coco/test-dev2017.txt
2024-01-31 16:17:11,298 [INFO] data.nc                                 3
2024-01-31 16:17:11,298 [INFO] data.names                              ['person', 'bicycle', 'car']
2024-01-31 16:17:11,298 [INFO] train_transforms.stage_epochs           [490, 10]
2024-01-31 16:17:11,298 [INFO] train_transforms.trans_list             [[{'func_name': 'mosaic', 'prob': 1.0}, {'func_name': 'resample_segments'}, {'func_name': 'random_perspective', 'prob': 1.0, 'degrees': 0.0, 'translate': 0.1, 'scale': 0.5, 'shear': 0.0}, {'func_name': 'albumentations'}, {'func_name': 'hsv_augment', 'prob': 1.0, 'hgain': 0.015, 'sgain': 0.7, 'vgain': 0.4}, {'func_name': 'fliplr', 'prob': 0.5}, {'func_name': 'label_norm', 'xyxy2xywh_': True}, {'func_name': 'label_pad', 'padding_size': 160, 'padding_value': -1}, {'func_name': 'image_norm', 'scale': 255.0}, {'func_name': 'image_transpose', 'bgr2rgb': True, 'hwc2chw': True}], [{'func_name': 'letterbox', 'scaleup': True}, {'func_name': 'resample_segments'}, {'func_name': 'random_perspective', 'prob': 1.0, 'degrees': 0.0, 'translate': 0.1, 'scale': 0.5, 'shear': 0.0}, {'func_name': 'albumentations'}, {'func_name': 'hsv_augment', 'prob': 1.0, 'hgain': 0.015, 'sgain': 0.7, 'vgain': 0.4}, {'func_name': 'fliplr', 'prob': 0.5}, {'func_name': 'label_norm', 'xyxy2xywh_': True}, {'func_name': 'label_pad', 'padding_size': 160, 'padding_value': -1}, {'func_name': 'image_norm', 'scale': 255.0}, {'func_name': 'image_transpose', 'bgr2rgb': True, 'hwc2chw': True}]]
2024-01-31 16:17:11,298 [INFO] data.test_transforms                    [{'func_name': 'letterbox', 'scaleup': False, 'only_image': True}, {'func_name': 'image_norm', 'scale': 255.0}, {'func_name': 'image_transpose', 'bgr2rgb': True, 'hwc2chw': True}]
2024-01-31 16:17:11,298 [INFO] data.num_parallel_workers               4
2024-01-31 16:17:11,298 [INFO] optimizer.optimizer                     momentum
2024-01-31 16:17:11,298 [INFO] optimizer.lr_init                       0.01
2024-01-31 16:17:11,298 [INFO] optimizer.momentum                      0.937
2024-01-31 16:17:11,298 [INFO] optimizer.nesterov                      True
2024-01-31 16:17:11,298 [INFO] optimizer.loss_scale                    1.0
2024-01-31 16:17:11,298 [INFO] optimizer.warmup_epochs                 3
2024-01-31 16:17:11,298 [INFO] optimizer.warmup_momentum               0.8
2024-01-31 16:17:11,298 [INFO] optimizer.warmup_bias_lr                0.1
2024-01-31 16:17:11,298 [INFO] optimizer.min_warmup_step               1000
2024-01-31 16:17:11,298 [INFO] optimizer.group_param                   yolov8
2024-01-31 16:17:11,298 [INFO] optimizer.gp_weight_decay               0.0005
2024-01-31 16:17:11,298 [INFO] optimizer.start_factor                  1.0
2024-01-31 16:17:11,298 [INFO] optimizer.end_factor                    0.01
2024-01-31 16:17:11,298 [INFO] optimizer.epochs                        500
2024-01-31 16:17:11,298 [INFO] optimizer.nbs                           64
2024-01-31 16:17:11,298 [INFO] optimizer.accumulate                    1
2024-01-31 16:17:11,298 [INFO] optimizer.total_batch_size              16
2024-01-31 16:17:11,298 [INFO] loss.name                               YOLOv8Loss
2024-01-31 16:17:11,298 [INFO] loss.box                                7.5
2024-01-31 16:17:11,298 [INFO] loss.cls                                0.5
2024-01-31 16:17:11,298 [INFO] loss.dfl                                1.5
2024-01-31 16:17:11,298 [INFO] loss.reg_max                            16
2024-01-31 16:17:11,298 [INFO] network.model_name                      yolov8
2024-01-31 16:17:11,298 [INFO] network.nc                              3
2024-01-31 16:17:11,298 [INFO] network.reg_max                         16
2024-01-31 16:17:11,298 [INFO] network.stride                          [8, 16, 32]
2024-01-31 16:17:11,298 [INFO] network.backbone                        [[-1, 1, 'ConvNormAct', [64, 3, 2]], [-1, 1, 'ConvNormAct', [128, 3, 2]], [-1, 3, 'C2f', [128, True]], [-1, 1, 'ConvNormAct', [256, 3, 2]], [-1, 6, 'C2f', [256, True]], [-1, 1, 'ConvNormAct', [512, 3, 2]], [-1, 6, 'C2f', [512, True]], [-1, 1, 'ConvNormAct', [1024, 3, 2]], [-1, 3, 'C2f', [1024, True]], [-1, 1, 'SPPF', [1024, 5]]]
2024-01-31 16:17:11,298 [INFO] network.head                            [[-1, 1, 'Upsample', ['None', 2, 'nearest']], [[-1, 6], 1, 'Concat', [1]], [-1, 3, 'C2f', [512]], [-1, 1, 'Upsample', ['None', 2, 'nearest']], [[-1, 4], 1, 'Concat', [1]], [-1, 3, 'C2f', [256]], [-1, 1, 'ConvNormAct', [256, 3, 2]], [[-1, 12], 1, 'Concat', [1]], [-1, 3, 'C2f', [512]], [-1, 1, 'ConvNormAct', [512, 3, 2]], [[-1, 9], 1, 'Concat', [1]], [-1, 3, 'C2f', [1024]], [[15, 18, 21], 1, 'YOLOv8Head', ['nc', 'reg_max', 'stride']]]
2024-01-31 16:17:11,298 [INFO] network.depth_multiple                  0.33
2024-01-31 16:17:11,298 [INFO] network.width_multiple                  0.5
2024-01-31 16:17:11,298 [INFO] network.max_channels                    1024
2024-01-31 16:17:11,298 [INFO] config                                  ./configs/yolov8/yolov8s.yaml
2024-01-31 16:17:11,298 [INFO] rank                                    0
2024-01-31 16:17:11,298 [INFO] rank_size                               1
2024-01-31 16:17:11,298 [INFO] total_batch_size                        16
2024-01-31 16:17:11,298 [INFO] callback                                []
2024-01-31 16:17:11,298 [INFO] 
2024-01-31 16:17:11,300 [INFO] Please check the above information for the configurations
2024-01-31 16:17:12,013 [WARNING] Parse Model, args: nearest, keep str type
2024-01-31 16:17:12,053 [WARNING] Parse Model, args: nearest, keep str type
2024-01-31 16:17:12,332 [INFO] number of network params, total: 11.156796M, trainable: 11.136745M
2024-01-31 16:18:12,847 [WARNING] Parse Model, args: nearest, keep str type
2024-01-31 16:18:12,888 [WARNING] Parse Model, args: nearest, keep str type
2024-01-31 16:18:13,176 [INFO] number of network params, total: 11.156796M, trainable: 11.136745M
2024-01-31 16:18:26,561 [INFO] ema_weight not exist, default pretrain weight is currently used.
2024-01-31 16:18:26,630 [INFO] Dataset Cache file hash/version check success.
2024-01-31 16:18:26,630 [INFO] Load dataset cache from [/home/ma-user/work/ljy/mindyolo-master/mindyolo/dataset/increment/increment_V1_total/train2017.cache.npy] success.

Scanning '/home/ma-user/work/ljy/mindyolo-master/mindyolo/dataset/increment/increment_V1_total/train2017.cache.npy' images and labels... 1819 found, 223 missing, 0 empty, 0 corrupted: 100%|██████████| 2042/2042 [00:00<?, ?it/s]
Scanning '/home/ma-user/work/ljy/mindyolo-master/mindyolo/dataset/increment/increment_V1_total/train2017.cache.npy' images and labels... 1819 found, 223 missing, 0 empty, 0 corrupted: 100%|██████████| 2042/2042 [00:00<00:00, 12539924.99it/s]
2024-01-31 16:18:26,667 [INFO] Dataloader num parallel workers: [4]
2024-01-31 16:18:26,743 [INFO] Dataset Cache file hash/version check success.
2024-01-31 16:18:26,743 [INFO] Load dataset cache from [/home/ma-user/work/ljy/mindyolo-master/mindyolo/dataset/increment/increment_V1_total/train2017.cache.npy] success.

Scanning '/home/ma-user/work/ljy/mindyolo-master/mindyolo/dataset/increment/increment_V1_total/train2017.cache.npy' images and labels... 1819 found, 223 missing, 0 empty, 0 corrupted: 100%|██████████| 2042/2042 [00:00<?, ?it/s]
Scanning '/home/ma-user/work/ljy/mindyolo-master/mindyolo/dataset/increment/increment_V1_total/train2017.cache.npy' images and labels... 1819 found, 223 missing, 0 empty, 0 corrupted: 100%|██████████| 2042/2042 [00:00<00:00, 14665699.95it/s]
2024-01-31 16:18:26,778 [INFO] Dataloader num parallel workers: [4]
2024-01-31 16:18:32,957 [INFO] Registry(name=callback, total=4)
2024-01-31 16:18:32,957 [INFO]   (0): YoloxSwitchTrain in mindyolo/utils/callback.py
2024-01-31 16:18:32,957 [INFO]   (1): EvalWhileTrain in mindyolo/utils/callback.py
2024-01-31 16:18:32,957 [INFO]   (2): SummaryCallback in mindyolo/utils/callback.py
2024-01-31 16:18:32,957 [INFO]   (3): ProfilerCallback in mindyolo/utils/callback.py
2024-01-31 16:18:32,957 [INFO] 
2024-01-31 16:18:33,861 [INFO] got 1 active callback as follows:
2024-01-31 16:18:33,862 [INFO] SummaryCallback()
2024-01-31 16:18:33,862 [WARNING] The first epoch will be compiled for the graph, which may take a long time; You can come back later :).
[WARNING] package not installed, albumentations load failed
[WARNING] package not installed, albumentations load failed
[WARNING] package not installed, albumentations load failed
[WARNING] package not installed, albumentations load failed
Traceback (most recent call last):
  File "/home/ma-user/work/ljy/mindyolo-master/train.py", line 321, in <module>
    train(args)
  File "/home/ma-user/work/ljy/mindyolo-master/train.py", line 276, in train
    trainer.train(
  File "/home/ma-user/work/ljy/mindyolo-master/mindyolo/utils/trainer_factory.py", line 153, in train
    for i, data in enumerate(loader):
  File "/home/ma-user/anaconda3/envs/MindSpore/lib/python3.9/site-packages/mindspore/dataset/engine/iterators.py", line 141, in __next__
    data = self._get_next()
  File "/home/ma-user/anaconda3/envs/MindSpore/lib/python3.9/site-packages/mindspore/dataset/engine/iterators.py", line 230, in _get_next
    raise err
  File "/home/ma-user/anaconda3/envs/MindSpore/lib/python3.9/site-packages/mindspore/dataset/engine/iterators.py", line 213, in _get_next
    return {k: self._transform_md_to_output(t) for k, t in self._iterator.GetNextAsMap().items()}
RuntimeError: Exception thrown from user defined Python function in dataset.

------------------------------------------------------------------
- Python Call Stack: 
------------------------------------------------------------------
Traceback (most recent call last):
  File "/home/ma-user/anaconda3/envs/MindSpore/lib/python3.9/site-packages/mindspore/dataset/engine/datasets_user_defined.py", line 285, in process
    yield _convert_row(result)
  File "/home/ma-user/anaconda3/envs/MindSpore/lib/python3.9/site-packages/mindspore/dataset/engine/datasets_user_defined.py", line 157, in _convert_row
    raise TypeError("The {}th item of input data is expected to be " \
TypeError: The 1th item of input data is expected to be int, float, str, bytes, numpy.ndarray, Tensor, but got dict.

------------------------------------------------------------------
- Dataset Pipeline Error Message: 
------------------------------------------------------------------
[ERROR] Execute user Python code failed, check 'Python Call Stack' above.

------------------------------------------------------------------
- C++ Call Stack: (For framework developers) 
------------------------------------------------------------------
mindspore/ccsrc/minddata/dataset/engine/datasetops/source/generator_op.cc(225).

[WARNING] package not installed, albumentations load failed
[WARNING] package not installed, albumentations load failed
[WARNING] package not installed, albumentations load failed
[WARNING] package not installed, albumentations load failed

ms版本为2.1.1时，日志为：
2024-02-02 11:29:11,623 [INFO] parse_args:
2024-02-02 11:29:11,623 [INFO] task                                    detect
2024-02-02 11:29:11,623 [INFO] device_target                           Ascend
2024-02-02 11:29:11,623 [INFO] save_dir                                ./runs/2024.02.02-11.29.11
2024-02-02 11:29:11,623 [INFO] device_per_servers                      8
2024-02-02 11:29:11,623 [INFO] log_level                               INFO
2024-02-02 11:29:11,623 [INFO] is_parallel                             False
2024-02-02 11:29:11,623 [INFO] ms_mode                                 0
2024-02-02 11:29:11,623 [INFO] ms_amp_level                            O0
2024-02-02 11:29:11,623 [INFO] keep_loss_fp32                          True
2024-02-02 11:29:11,623 [INFO] ms_loss_scaler                          static
2024-02-02 11:29:11,623 [INFO] ms_loss_scaler_value                    1024.0
2024-02-02 11:29:11,623 [INFO] ms_jit                                  True
2024-02-02 11:29:11,623 [INFO] ms_enable_graph_kernel                  False
2024-02-02 11:29:11,623 [INFO] ms_datasink                             False
2024-02-02 11:29:11,623 [INFO] overflow_still_update                   True
2024-02-02 11:29:11,623 [INFO] clip_grad                               False
2024-02-02 11:29:11,623 [INFO] clip_grad_value                         10.0
2024-02-02 11:29:11,623 [INFO] ema                                     True
2024-02-02 11:29:11,623 [INFO] weight                                  
2024-02-02 11:29:11,623 [INFO] ema_weight                              
2024-02-02 11:29:11,623 [INFO] freeze                                  []
2024-02-02 11:29:11,623 [INFO] epochs                                  500
2024-02-02 11:29:11,623 [INFO] per_batch_size                          16
2024-02-02 11:29:11,623 [INFO] img_size                                640
2024-02-02 11:29:11,623 [INFO] nbs                                     64
2024-02-02 11:29:11,623 [INFO] accumulate                              1
2024-02-02 11:29:11,623 [INFO] auto_accumulate                         False
2024-02-02 11:29:11,623 [INFO] log_interval                            100
2024-02-02 11:29:11,623 [INFO] single_cls                              False
2024-02-02 11:29:11,623 [INFO] sync_bn                                 False
2024-02-02 11:29:11,623 [INFO] keep_checkpoint_max                     100
2024-02-02 11:29:11,623 [INFO] run_eval                                False
2024-02-02 11:29:11,623 [INFO] conf_thres                              0.001
2024-02-02 11:29:11,623 [INFO] iou_thres                               0.7
2024-02-02 11:29:11,623 [INFO] conf_free                               True
2024-02-02 11:29:11,623 [INFO] rect                                    False
2024-02-02 11:29:11,623 [INFO] nms_time_limit                          20.0
2024-02-02 11:29:11,623 [INFO] recompute                               False
2024-02-02 11:29:11,623 [INFO] recompute_layers                        0
2024-02-02 11:29:11,623 [INFO] seed                                    2
2024-02-02 11:29:11,623 [INFO] summary                                 True
2024-02-02 11:29:11,623 [INFO] profiler                                False
2024-02-02 11:29:11,623 [INFO] profiler_step_num                       1
2024-02-02 11:29:11,623 [INFO] opencv_threads_num                      0
2024-02-02 11:29:11,623 [INFO] strict_load                             True
2024-02-02 11:29:11,623 [INFO] enable_modelarts                        False
2024-02-02 11:29:11,623 [INFO] data_url                                
2024-02-02 11:29:11,623 [INFO] ckpt_url                                
2024-02-02 11:29:11,623 [INFO] multi_data_url                          
2024-02-02 11:29:11,623 [INFO] pretrain_url                            
2024-02-02 11:29:11,623 [INFO] train_url                               
2024-02-02 11:29:11,623 [INFO] data_dir                                /cache/data/
2024-02-02 11:29:11,623 [INFO] ckpt_dir                                /cache/pretrain_ckpt/
2024-02-02 11:29:11,623 [INFO] data.dataset_name                       shwd
2024-02-02 11:29:11,623 [INFO] data.train_set                          /home/ma-user/work/ljy/mindyolo/dataset/SHWD/train.txt
2024-02-02 11:29:11,623 [INFO] data.val_set                            /home/ma-user/work/ljy/mindyolo/dataset/SHWD/val.txt
2024-02-02 11:29:11,623 [INFO] data.test_set                           /home/ma-user/work/ljy/mindyolo/dataset/SHWD/val.txt
2024-02-02 11:29:11,623 [INFO] data.nc                                 2
2024-02-02 11:29:11,623 [INFO] data.names                              ['person', 'hat']
2024-02-02 11:29:11,623 [INFO] train_transforms.stage_epochs           [490, 10]
2024-02-02 11:29:11,623 [INFO] train_transforms.trans_list             [[{'func_name': 'mosaic', 'prob': 1.0}, {'func_name': 'resample_segments'}, {'func_name': 'random_perspective', 'prob': 1.0, 'degrees': 0.0, 'translate': 0.1, 'scale': 0.5, 'shear': 0.0}, {'func_name': 'albumentations'}, {'func_name': 'hsv_augment', 'prob': 1.0, 'hgain': 0.015, 'sgain': 0.7, 'vgain': 0.4}, {'func_name': 'fliplr', 'prob': 0.5}, {'func_name': 'label_norm', 'xyxy2xywh_': True}, {'func_name': 'label_pad', 'padding_size': 160, 'padding_value': -1}, {'func_name': 'image_norm', 'scale': 255.0}, {'func_name': 'image_transpose', 'bgr2rgb': True, 'hwc2chw': True}], [{'func_name': 'letterbox', 'scaleup': True}, {'func_name': 'resample_segments'}, {'func_name': 'random_perspective', 'prob': 1.0, 'degrees': 0.0, 'translate': 0.1, 'scale': 0.5, 'shear': 0.0}, {'func_name': 'albumentations'}, {'func_name': 'hsv_augment', 'prob': 1.0, 'hgain': 0.015, 'sgain': 0.7, 'vgain': 0.4}, {'func_name': 'fliplr', 'prob': 0.5}, {'func_name': 'label_norm', 'xyxy2xywh_': True}, {'func_name': 'label_pad', 'padding_size': 160, 'padding_value': -1}, {'func_name': 'image_norm', 'scale': 255.0}, {'func_name': 'image_transpose', 'bgr2rgb': True, 'hwc2chw': True}]]
2024-02-02 11:29:11,623 [INFO] data.test_transforms                    [{'func_name': 'letterbox', 'scaleup': False, 'only_image': True}, {'func_name': 'image_norm', 'scale': 255.0}, {'func_name': 'image_transpose', 'bgr2rgb': True, 'hwc2chw': True}]
2024-02-02 11:29:11,623 [INFO] data.num_parallel_workers               4
2024-02-02 11:29:11,623 [INFO] optimizer.optimizer                     momentum
2024-02-02 11:29:11,623 [INFO] optimizer.lr_init                       0.01
2024-02-02 11:29:11,623 [INFO] optimizer.momentum                      0.937
2024-02-02 11:29:11,623 [INFO] optimizer.nesterov                      True
2024-02-02 11:29:11,623 [INFO] optimizer.loss_scale                    1.0
2024-02-02 11:29:11,623 [INFO] optimizer.warmup_epochs                 3
2024-02-02 11:29:11,623 [INFO] optimizer.warmup_momentum               0.8
2024-02-02 11:29:11,623 [INFO] optimizer.warmup_bias_lr                0.1
2024-02-02 11:29:11,623 [INFO] optimizer.min_warmup_step               1000
2024-02-02 11:29:11,623 [INFO] optimizer.group_param                   yolov8
2024-02-02 11:29:11,623 [INFO] optimizer.gp_weight_decay               0.0005
2024-02-02 11:29:11,623 [INFO] optimizer.start_factor                  1.0
2024-02-02 11:29:11,623 [INFO] optimizer.end_factor                    0.01
2024-02-02 11:29:11,623 [INFO] optimizer.epochs                        500
2024-02-02 11:29:11,623 [INFO] optimizer.nbs                           64
2024-02-02 11:29:11,623 [INFO] optimizer.accumulate                    1
2024-02-02 11:29:11,623 [INFO] optimizer.total_batch_size              16
2024-02-02 11:29:11,623 [INFO] loss.name                               YOLOv8Loss
2024-02-02 11:29:11,623 [INFO] loss.box                                7.5
2024-02-02 11:29:11,623 [INFO] loss.cls                                0.5
2024-02-02 11:29:11,623 [INFO] loss.dfl                                1.5
2024-02-02 11:29:11,623 [INFO] loss.reg_max                            16
2024-02-02 11:29:11,623 [INFO] network.model_name                      yolov8
2024-02-02 11:29:11,623 [INFO] network.nc                              2
2024-02-02 11:29:11,623 [INFO] network.reg_max                         16
2024-02-02 11:29:11,623 [INFO] network.stride                          [8, 16, 32]
2024-02-02 11:29:11,623 [INFO] network.backbone                        [[-1, 1, 'ConvNormAct', [64, 3, 2]], [-1, 1, 'ConvNormAct', [128, 3, 2]], [-1, 3, 'C2f', [128, True]], [-1, 1, 'ConvNormAct', [256, 3, 2]], [-1, 6, 'C2f', [256, True]], [-1, 1, 'ConvNormAct', [512, 3, 2]], [-1, 6, 'C2f', [512, True]], [-1, 1, 'ConvNormAct', [1024, 3, 2]], [-1, 3, 'C2f', [1024, True]], [-1, 1, 'SPPF', [1024, 5]]]
2024-02-02 11:29:11,623 [INFO] network.head                            [[-1, 1, 'Upsample', ['None', 2, 'nearest']], [[-1, 6], 1, 'Concat', [1]], [-1, 3, 'C2f', [512]], [-1, 1, 'Upsample', ['None', 2, 'nearest']], [[-1, 4], 1, 'Concat', [1]], [-1, 3, 'C2f', [256]], [-1, 1, 'ConvNormAct', [256, 3, 2]], [[-1, 12], 1, 'Concat', [1]], [-1, 3, 'C2f', [512]], [-1, 1, 'ConvNormAct', [512, 3, 2]], [[-1, 9], 1, 'Concat', [1]], [-1, 3, 'C2f', [1024]], [[15, 18, 21], 1, 'YOLOv8Head', ['nc', 'reg_max', 'stride']]]
2024-02-02 11:29:11,623 [INFO] network.depth_multiple                  0.33
2024-02-02 11:29:11,623 [INFO] network.width_multiple                  0.5
2024-02-02 11:29:11,623 [INFO] network.max_channels                    1024
2024-02-02 11:29:11,623 [INFO] config                                  ./configs/yolov8/yolov8s.yaml
2024-02-02 11:29:11,623 [INFO] rank                                    0
2024-02-02 11:29:11,623 [INFO] rank_size                               1
2024-02-02 11:29:11,623 [INFO] total_batch_size                        16
2024-02-02 11:29:11,623 [INFO] callback                                []
2024-02-02 11:29:11,623 [INFO] 
2024-02-02 11:29:11,626 [INFO] Please check the above information for the configurations
Segmentation fault (core dumped)

### 备注

MindSpore Lab/mindyolo

内容风险标识

评论 (0)

MindSpore Lab/mindyolo .gitee-modal { width: 500px !important; }

内容风险标识