videturfortuna / vehicle_reid_itsc2023

30 stars 4 forks source link

[Models] Problem about LAI Layer #12

Closed 22ema closed 6 months ago

22ema commented 6 months ago

When creating a self.LAI tensor, if nviews is 0 or ncams is 0, a tensor array of size (len(n_branches), 0, 2048) is created, causing a problem.

So I fix a problem. https://github.com/videturfortuna/vehicle_reid_itsc2023/pull/11#issue-2183387292

Could you please check the pull request above?