finetune pose-5b out of memory

[junwuzhang19]

Thanks for your great work! I finetune pose-5b using a single A100 80G using train_control.sh, but got out of memory. The training script is unchanged other than the model size. I wonder how to finetune pose-5b model on 80G GPU. thanks!

[yunkchen]

Another machine with more than one 80G GPU is all you need.