arcface-pytorch
arcface-pytorch copied to clipboard
这是一个arcface-pytorch的源码,可以用于训练自己的模型。
Load weights model_data/mobilenet_v1_backbone_weights.pth. Successful Load Key: [] …… Successful Load Key Num: 0 Fail To Load Key: ['stage1.0.0.weight', 'stage1.0.1.weight', 'stage1.0.1.bias', 'stage1.0.1.running_mean', 'stage1.0.1.running_var', 'stage1.0.1.num_batches_tracked', 'stage1.1.0.weight', 'stage1.1.1.weight', 'stage1.1.1.bias', 'stage1.1.1.running_mean', 'stage1.1.1.running_var', 'stage1.1.1.num_batches_tracked', 'stage1.1.3.weight',...
你好,请问能实现下转onnx和onnx推理的功能吗
您好,请问为什么我把backbone换成swin transformer,并换了成了我自己的数据集之后,训练速度及其慢,我用单卡4090进行训练,六个小时才跑完1个epoch。 Epoch 1/100: 100%|███████████████████████████████████████████████████████████████| 5287/5287 [6:08:50