daimuuc / PyramidScaleNetwork

To the best of our knowledge, this is the first work to explicitly address feature similarity issue in multi-column design. Extensive experiments on four challenging benchmarks (ShanghaiTech, UCF_CC_50, UCF-QNRF, and Mall) demonstrate the effectiveness of the proposed innovations as well as the superior performance over the state-of-the-art. More remarkably, our method can be easily applied to other existing multi-column models as a plug-in to significantly boost the performance to a large extent.
18 stars 2 forks source link

The problem of training on UCF-QNRF #1

Open 645873229 opened 3 years ago

645873229 commented 3 years ago

I have set the batch size to 1, but the problem of “CUDA out of memory Tried to allocate 38.49 GiB " still occurs when training on UCF-QNRF dataset

zuofengyouliang commented 1 year ago

我也有同样的问题,请问您解决了吗