arcface-pytorch icon indicating copy to clipboard operation
arcface-pytorch copied to clipboard

关于net/arcface.py文件第34行

Open Joseph-Winkle opened this issue 3 years ago • 3 comments

我在租的服务器上跑的时候遇到了“RuntimeError: expected device cuda:0 and dtype Float but got device cuda:0 and dtype Long”这个错误, torch是1.2.0版本的,网上说是因为版本太低,也可以通过更改数据类型解决, 我是把onehot替换为onehot.float()之后就解决了。

Joseph-Winkle avatar May 18 '22 03:05 Joseph-Winkle

0 0我这里没遇到诶,你具体的版本就是1.2吗,可是我尝试的环境也是1.2诶

bubbliiiing avatar May 18 '22 06:05 bubbliiiing

我也觉得奇怪,不知道是不是平台的原因

Joseph-Winkle avatar May 18 '22 06:05 Joseph-Winkle

0 0迷起来了

bubbliiiing avatar May 19 '22 09:05 bubbliiiing