search

RubickH / Image-Captioning-with-MAD-and-SAP

Code for paper "Image Captioning with End-to-End Attribute Detection and Subsequent Attributes Prediction". IEEE Transactions on Image Processing 2020
https://ieeexplore.ieee.org/document/8976408
26 stars 4 forks source link

Training Time #5

Open demraf opened 3 years ago

demraf commented 3 years ago

How long it takes to train this model with using 1 GPU machine?

RubickH commented 3 years ago

It depends on the hardwares like GPU, cpu and RAM, etc. Personally, I suggest that it's better not use only 1 gpu to train captioning models. It will take several days to train a model on MSCOCO. huang-yq17 @.*** 签名由 网易邮箱大师 定制 On 07/01/2021 16:36, demraf wrote: How long it takes to train this model with using 1 GPU machine? — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or unsubscribe.