yang-0201 / yolov6_pro Goto Github PK

Model Summary: 550 layers, 13156006 parameters, 13155972 gradients
Loading state_dict from weights/yolov6t_yaml_new.pt for fine-tuning...
backbone.10.conv.weightcan not change
backbone.10.bn.weightcan not change
backbone.10.bn.biascan not change
backbone.10.bn.running_meancan not change
backbone.10.bn.running_varcan not change
backbone.14.conv.weightcan not change
backbone.14.bn.weightcan not change
backbone.14.bn.biascan not change
backbone.14.bn.running_meancan not change
backbone.14.bn.running_varcan not change
******************
transform_weight: 314/668 from weights/yolov6t_yaml_new.pt
transform_model_weight: 314/768 from weights/yolov6t_yaml_new.pt
******************
Training start...

     Epoch  iou_loss  dfl_loss  cls_loss
  0%|          | 0/186 [00:00<?, ?it/s]                                                                                                                                      /root/miniconda3/envs/py38/lib/python3.8/site-packages/torch/functional.py:445: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at  ../aten/src/ATen/native/TensorShape.cpp:2157.)
  return _VF.meshgrid(tensors, **kwargs)  # type: ignore[attr-defined]
  0%|          | 0/186 [00:00<?, ?it/s]                                                                                                                                      
ERROR in training steps.
ERROR in training loop or eval/save model.
Traceback (most recent call last):
  File "tools/train.py", line 127, in <module>
    main(args)
  File "tools/train.py", line 117, in main
    trainer.train()
  File "/root/autodl-nas/YOLOv6_pro-main/yolov6/core/engine.py", line 99, in train
    self.train_in_loop(self.epoch)
  File "/root/autodl-nas/YOLOv6_pro-main/yolov6/core/engine.py", line 113, in train_in_loop
    self.train_in_steps(epoch_num, self.step)
  File "/root/autodl-nas/YOLOv6_pro-main/yolov6/core/engine.py", line 142, in train_in_steps
    total_loss, loss_items = self.compute_loss(preds, targets, epoch_num, step_num)
  File "/root/autodl-nas/YOLOv6_pro-main/yolov6/models/loss.py", line 155, in __call__
    loss_cls = self.varifocal_loss(pred_scores, target_scores, one_hot_label)
  File "/root/miniconda3/envs/py38/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
    return forward_call(*input, **kwargs)
  File "/root/autodl-nas/YOLOv6_pro-main/yolov6/models/loss.py", line 196, in forward
    weight = alpha * pred_score.pow(gamma) * (1 - label) + gt_score * label
RuntimeError: The size of tensor a (8) must match the size of tensor b (4) at non-singleton dimension 2