search

mobulan / IELT

Source code of the paper Fine-Grained Visual Classification via Internal Ensemble Learning Transformer
MIT License
39 stars 4 forks source link

y.cuda() #2

Closed RC-Qiao closed 1 year ago

RC-Qiao commented 1 year ago

我在训我自己的数据集时发现y是list无法放入cuda。我改变了y的数据类型但是在交叉熵损失又出现了错误，请问你训flowers的y是什么类型。我该如何解决这个问题