OOM Problem

[DavisMeee]

Many thanks for releasing this interesting work.
I would like to know the minimum GPU requirements for training. I am using a 48GB GPU but still have an Cuda out-of-memory (OOM) problem. I saw you used A100 for training, so I would like to ask for further details about it.

[gongzix]

We are using a single 80G A100, so you may need to adjust the batch size.