torch.cuda.OutOfMemoryError for A5000

[Jamie725]

Hi authors,

Thanks for the great work!

From the README file, it says A5000 should work for the codes.

However, I tried the codes with A5000 but encountered out-of-memory issue for A5000.

I was wondering if there are some adjustments from the code so that A5000 doesn't work anymore.

Or, is there anything wrong with my installation that could result in the difference?

Thanks a lot for your help!