Memory issues with the earthformer runtime

[fizzking]

I set the dimension as (2000,131,21,1), that is, (T,H,W,C) the data input model can run, (the monitor shows that the runtime code takes up 13GB), but when increasing the timing, that is, expanding to (4000,131,21,1), the memory is insufficient. My GPU is 24GB and the batch size is set to 2, I wonder if this is because the earthformer model itself takes up a lot of memory at runtime?

[gaozhihan]

As the shape of the data increases, the memory costs also increase.