Comparison exception: The values for attribute 'shape' do not match: torch.Size([]) != torch.Size([1, 8400]).

microsoft / nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

https://nni.readthedocs.io

MIT License

13.88k stars 1.81k forks source link

Comparison exception: The values for attribute 'shape' do not match: torch.Size([]) != torch.Size([1, 8400]). #5741

Open Gooddz1 opened 5 months ago

Gooddz1 commented 5 months ago

使用nni对yolov8进行剪枝，报了如下错误 Comparison exception: The values for attribute 'shape' do not match: torch.Size([]) != torch.Size([1, 8400]).