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

Published 20 hours ago • verified publisher icon bubbliiiing

训练速度极慢

Open 20191864218 opened this issue 5 months ago • 0 comments

您好，请问为什么我把backbone换成swin transformer，并换了成了我自己的数据集之后，训练速度及其慢，我用单卡4090进行训练，六个小时才跑完1个epoch。 Epoch 1/100: 100%|███████████████████████████████████████████████████████████████| 5287/5287 [6:08:50<00:00, 4.19s/it, accuracy=0.264, lr=0.0001, total_loss=9.2]

20191864218 avatar Jul 03 '25 00:07 20191864218