请教yolov6训练报错

Question

Training start...
C:\Users\11754\.conda\envs\yoyo\lib\site-packages\torch\cuda\amp\grad_scaler.py:115: UserWarning: torch.cuda.amp.GradScaler is enabled, but CUDA is not available.  Disabling.
  warnings.warn("torch.cuda.amp.GradScaler is enabled, but CUDA is not available.  Disabling.")
ERROR in training loop or eval/save model.

Training completed in 0.000 hours.
Traceback (most recent call last):
  File "J:\dll\yolo6\YOLOv6-0.2.0\yolov6\core\engine.py", line 97, in train
    self.train_before_loop()
  File "J:\dll\yolo6\YOLOv6-0.2.0\yolov6\core\engine.py", line 226, in train_before_loop
    distill_weight = self.cfg.model.head.distill_weight,
  File "J:\dll\yolo6\YOLOv6-0.2.0\yolov6\utils\config.py", line 30, in __getattr__
    raise ex
AttributeError: 'ConfigDict' object has no attribute 'distill_weight'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "J:\dll\yolo6\YOLOv6-0.2.0\tools\train.py", line 126, in <module>
    main(args)
  File "J:\dll\yolo6\YOLOv6-0.2.0\tools\train.py", line 116, in main
    trainer.train()
  File "J:\dll\yolo6\YOLOv6-0.2.0\yolov6\core\engine.py", line 105, in train
    self.train_after_loop()
  File "J:\dll\yolo6\YOLOv6-0.2.0\yolov6\core\engine.py", line 261, in train_after_loop
    strip_optimizer(save_ckpt_dir, self.epoch)  # strip optimizers for saved pt model
AttributeError: 'Trainer' object has no attribute 'epoch'

AttributeError: 'ConfigDict' object has no attribute 'distill_weight'

During handling of the above exception, another exception occurred:

AttributeError: 'Trainer' object has no attribute 'epoch' 一直提示这个 我都pip安装了运行还是提示 不知道哪点问题

Answer 1

我记得之前我看issue，美团官方是提了一个PR，解决了这个bug，如果你这里的代码是最新版本的还是有问题，请联系美团官方。

Resume Training issue: AttributeError: 'Trainer' object has no attribute 'epoch' · Issue #212 · meituan/YOLOv6 · GitHub Hello, Trying to resume training on a custom dataset and got this error: strip_optimizer(save_ckpt_dir, self.epoch) # strip optimizers for saved pt model AttributeError: 'Trainer' object has no attribute 'epoch' https://github.com/meituan/YOLOv6/issues/212