KaihuaTang / VQA2.0-Recent-Approachs-2018.pytorch

A pytroch reimplementation of "Bilinear Attention Network", "Intra- and Inter-modality Attention", "Learning Conditioned Graph Structures", "Learning to count object", "Bottom-up top-down" for Visual Question Answering 2.0
GNU General Public License v3.0
295 stars 57 forks source link

something about training #5

Closed IRIS00123 closed 4 years ago

IRIS00123 commented 5 years ago

您好,在GitHub上看到了您对近年来VQA模型复现的Git项目,非常感兴趣。我是一名对VQA感兴趣的在读研究生,有个问题想请教一下,在训练过程中,dataloader部分,question和answer的索引和编码占用了大约两至三分钟的时间,导致训练一个epoch需要大量的时间,训练效率低。您是怎么解决这个问题的呢?有什么建议吗?万分感谢![机智]

KaihuaTang commented 5 years ago

应该是服务器内存不够大,所以卡在IO上了