It takes too long to train

CCIIPLab / DPT

The code of IJCAI2022 paper, Declaration-based Prompt Tuning for Visual Question Answering

19 stars 2 forks source link

It takes too long to train #7

Closed zhongfansun closed 1 year ago

zhongfansun commented 1 year ago

Thank you for providing your project. But I need about 6 hours to train one epoch of MLM tasks even if _per_gpu_train_batchsize is halved to 16. I ran on the single 3090. Is this normal? Look forward to your reply.