search

KunpengLi1994 / VSRN

PyTorch code for ICCV'19 paper "Visual Semantic Reasoning for Image-Text Matching"
288 stars 47 forks source link

train coco memory out #19

Closed shuihuayi closed 4 years ago

shuihuayi commented 4 years ago

when I train on the coco dataset is always memory out of range , but it is fine when train on the flick30k, how do you deal this problem . thanks .

shuihuayi commented 4 years ago

I find that I need to swap