May I ask how much memory is required for IPG training? I encounter an OOM error when running the training script on A100-40G cards.