AttributeError: ‘list‘ object has no attribute ‘view‘

问题描述

训练yolov9的时候遇到了下面的问题。

In loss_tal.py: pred_distri, pred_scores = torch.cat([xi.view(feats[0].shape[0], self.no, -1) for xi in feats], 2).split(
(self.reg_max * 4, self.nc), 1)
The error is as follows：
AttributeError: 'list' object has no attribute 'view'

解决方法

使用train_dual.py脚本训练模型。作者回答的，必须正确！！！

解决后的效果

这是暗示我上班要996吗？