关于model.py

[Arlen-yuzu]

您好，请问在model.py的ImageEncoder中，foward()内的这两句代码是什么意思？ features_top = features_orig[-1] features = features_top.view(features_top.size(0), features_top.size(1), -1).transpose(2, 1) # b, 49, 2048

[Arlen-yuzu]

请问是否需要在训练部分对代码进行调整，我在训练完2个epoch后就会报 CUDA out of memory 的错误，但是我是11G的2080，不知道是什么原因出现这样的问题.

[kywen1119]

是在evaluate的时候产生了oom，你可以尝试调低train或者eval的batch size

[jasscia18]

请问您训练COCO和f30k数据集花了多长时间？

[kywen1119]

coco12h左右，f30k 4h左右