Something about OOM

[seriouschenwannachange-art]

Hello!
When I run this model (using an A800 graphics card with 80GB of video memory), I found OOM errors in the level layers aggregation and de aggregation (in the encoder and decoder part). So I checked the memory usage and found that the aggregation and de aggregation steps were very memory intensive. Do you have any solutions?

[wesselb]

Hey @seriouschenwannachange-art! Thanks for opening an issue. Unfortunately the model is memory intensive. There are no simple solutions. It's probably best to start with ensuring that your environment is optimised. See this section of the docs.

[seriouschenwannachange-art]

@wesselb Thanks for your answer! I will think about how to solve this problem.