How long does it take to train this model?

[TongkunGuan]

It's a good work! I would like to know how long it will take to train this model.

[ZYM-PKU]

Thanks. Under current implementation, it takes about 48 hrs to train on the LAION-OCR dataset (batchsize=64, iters=100k, gpu=NVIDIA A40)

[TongkunGuan]

Thanks. Under current implementation, it takes about 48 hrs to train on the LAION-OCR dataset (batchsize=64, iters=100k, gpu=NVIDIA A40)

Thanks for your reply！ How many GPUs are needed for 48h training?

[ZYM-PKU]

I used 8*A40