About memory issue

[YamamotoSayaka]

Hi @zhanghaicang ,thanks again for sharing this code.
I'm also trying to reimplement his code now.
However, I'm having a memory issue here, 12GB of GPU memory is never enough for me.
Operating directly on a L*L map with no pooling looks unacceptable.
Any advice?

[YamamotoSayaka]

And how about the performance? Are you able to replicate the results in the paper?
Many thx!

[zhanghaicang]

Hi @zhanghaicang ,thanks again for sharing this code.
I'm also trying to reimplement his code now.
However, I'm having a memory issue here, 12GB of GPU memory is never enough for me.
Operating directly on a L*L map with no pooling looks unacceptable.
Any advice?

You could try a smaller batch size.

[YamamotoSayaka]

Hi, @zhanghaicang sorry to bother again.
How about performance?
And about the input, the author provided a pickle file from their raptorX website
However, from your code, you are using a different csv file, can you kindly tell me the difference?