training GPU configuration

[CrossLee1]

Thanks for your excellent repo.

When training cc12m_32x1024 with type VitGAN or MLP Mixer, what kinds of GPU environment do you use? Tesla V100 with 32G mem or others?

Thanks

[mehdidc]

So I guess similar answer here @CrossLee1, see issue #21. For the GPU it's A100s with 40GBs.