arcface-pytorch
arcface-pytorch copied to clipboard
关于net/arcface.py文件第34行
我在租的服务器上跑的时候遇到了“RuntimeError: expected device cuda:0 and dtype Float but got device cuda:0 and dtype Long”这个错误, torch是1.2.0版本的,网上说是因为版本太低,也可以通过更改数据类型解决, 我是把onehot替换为onehot.float()之后就解决了。
0 0我这里没遇到诶,你具体的版本就是1.2吗,可是我尝试的环境也是1.2诶
我也觉得奇怪,不知道是不是平台的原因
0 0迷起来了