RuntimeError: Error(s) in loading state_dict for ResNet50:size mismatch for classifier.weight

问题思路

加载的模型的分类器和现在训练的分类不一致

解决问题

把分类器那里改成一样的就好了

# model = models.init_model(name=args.arch, num_classes=dataset.num_train_pids, loss={'xent', 'htri'})
    model = models.init_model(name=args.arch, num_classes=767, loss={'xent', 'htri'})

这种改法就比较简单粗暴了，可能会有更巧妙的解决方案