maywander

model-zoo里有 找insightface预训练模型能找到 在 2019-08-16 18:59:29，"liutt1993" 写道： 请问这里面有训练过的模型吗？我想直接用来提取特征在哪得到?谢谢 — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or mute...

So have you solved the problem？can you share the difference between setting tta=true and tta=false

我遇到了同样的问题，发现问题的原因是数据并行和optim.SGD顺序颠倒。 当编写为 model=Backbone() if torch.cuda.device_count() > 1: model = nn.DataParallel(model,device_ids=[0,1,2,3] model.to(conf.device) optimizer = optim.SGD(,,,,,,,) 发生报错 ———————— 当改成下面结构后报错消失 model=Backbone() optimizer = optim.SGD(,,,,,,,) if torch.cuda.device_count() > 1: model = nn.DataParallel(model,device_ids=[0,1,2,3] model.to(conf.device)