问题描述:
dataset的get_item()是返回了 词向量inp,img,label。
最初是这么写的:
return (img, 词向量),target
这样的话在送入model的时候,然后报错:list是无法放到cuda上。
解决:
所以采用将每个返回值张量化,根据longtensor()或floattensor()进行返回
return longtensor(img),longtensor(inp),longtensor(target)
这样就能在cuda上使用了。
问题描述:
dataset的get_item()是返回了 词向量inp,img,label。
最初是这么写的:
return (img, 词向量),target
这样的话在送入model的时候,然后报错:list是无法放到cuda上。
解决:
所以采用将每个返回值张量化,根据longtensor()或floattensor()进行返回
return longtensor(img),longtensor(inp),longtensor(target)
这样就能在cuda上使用了。