問題描述
訓練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嗎?