ERROR while evaling：expected 4-dimensional weight, but got 6-dimensional input of size instead

wzzheng / GenAD

[ECCV 2024] GenAD: Generative End-to-End Autonomous Driving

Apache License 2.0

151 stars 14 forks source link

ERROR while evaling：expected 4-dimensional weight, but got 6-dimensional input of size instead #12

Open zihan912 opened 1 week ago

zihan912 commented 1 week ago

I have pretraiend the model for 60epochs, and want to use test.py to test and eval. However I got this error, could anyone help me solve this problem? 微信图片_20240625163548