search

kuanghuei / SCAN

PyTorch source code for "Stacked Cross Attention for Image-Text Matching" (ECCV 2018)
Apache License 2.0
549 stars 113 forks source link

GPU training #39

Closed abhidipbhattacharyya closed 4 years ago

abhidipbhattacharyya commented 4 years ago

Hello, I am trying to train this model using GPU. I am reading your code. Have you used gpu for training? I am unable to find any .to(device) or .cuda(). Am I missing something?

kuanghuei commented 4 years ago

Do you mean like this line: https://github.com/kuanghuei/SCAN/blob/master/model.py#L437 ?