Video memory usage

[BarryGustin]

Hello, what is the minimum amount of video memory required for your training process?

[Haiyang-W]

The code is a bit old. I may not remember it clearly. We trained it using a 40GB A100, but if you reduce the batch size, for example, to 2, it should also work on a 20GB GPU.