arcface-pytorch

arcface-pytorch copied to clipboard

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)} KeyError: 'stage1.0.0.weight'

Let me see your code ?

What you change

解决了。

训练的accuracy和loss能到多少？

下面是我训练的数据，accuracy和loss很难改善。。。 Start Train Epoch 91/200: 100%|██████████| 7006/7006 [24:52<00:00, 4.69it/s, accuracy=0.639, lr=2.7e-5, total_loss=3.68] Finish Train Start Validation Epoch 91/200: 100%|██████████| 70/70 [00:05<00:00, 13.15it/s, accuracy=0.504, lr=2.7e-5, total_loss=8.59] Finish Validation 开始进行LFW数据集的验证。 LFW_Accuracy: 0.98733+-0.00611 Total Loss: 3.6754

0 0accuracy？ 没注意，我开了dropout，acc会低一些的

问一下backbone用mobilenet， 从0开始训练要多久呀？你用的什么硬件配置？ 我只有一个GPU，感觉训练很慢。。。都没信心了。

0 0accuracy？ 没注意，我开了dropout，acc会低一些的

不是0， 60%+， accuracy提升很慢，loss下降也非常慢。

98不是很高了吗

开始进行LFW数据集的验证。 LFW_Accuracy: 0.98733+-0.00611

训练的精度比较低

accuracy=0.639, lr=2.7e-5, total_loss=3.68

那个accuracy无用得。