arcface-pytorch
arcface-pytorch copied to clipboard
bubbliiiing
这是一个arcface-pytorch的源码，可以用于训练自己的模型。
Hi, How can I convert those pth models to onnx models? I tried several torch2onnx.py (https://github.com/deepinsight/insightface/blob/master/recognition/arcface_torch/torch2onnx.py) from different repositories but fails. Many thanks! -Scott
我在租的服务器上跑的时候遇到了“RuntimeError: expected device cuda:0 and dtype Float but got device cuda:0 and dtype Long”这个错误， torch是1.2.0版本的，网上说是因为版本太低，也可以通过更改数据类型解决， 我是把onehot替换为onehot.float()之后就解决了。
use my own datasets,however,the acc of train.py always equals 0.At the same time,the loss function does not converge.Why? i need adjust hyper-parameter(s or m)?
use this weights to train. but unfortunately, it raised an error as below: arcface-pytorch/train.py", line 144, in pretrained_dict = {k: v for k, v in pretrained_dict.items() if np.shape(model_dict[k]) == np.shape(v)}...
您好，不好意思打扰您了，想请问一下为什么加载了您提供的预训练的model以后，在CASIA-WebFace数据集训练时，刚开始的准确率还是比较低呢（从0开始）？正常来说您的模型是在CASIA-WebFace数据集训练的，那么一开始的准确率也应该比较高吧。
感谢博主开源！从读研开始一直关注博主，谢谢博主贡献代码。 下边是我运行时候遇到的两个问题，如果有遇到同样问题可以参考。 在运行时候会遇到两个问题，在win系统上，会报错ValueError: setting an array element with a sequence. The requested array has an inhomogeneous shape after 1 dimensions. The detected shape was (6000,) + inhomogeneous part（pytorch版本为2.0.1）.linux系统上会报错RuntimeError: CuDNN error:...
