If loading model weights with Error....

[shenyingying]

@cinastanbean
thanks for your great job,but i can't find the model_best_checkpoint_resnet18.pth.tar ,it's different with it,can you post this weights,looking for your replay,thanks very much.

[cinastanbean]

Hi, 这个不是标准的resnet18, 它的头部被更改过以用于解决多任务分类问题；

self.model.fc = mlmcmodel.BuildMultiLabelModel(fc_features, class_numbers)

ref: https://github.com/cinastanbean/pytorch-onnx-tensorflow-pb/blob/master/diymodel.py

你可以修改这一部分，然后加载官方的权重。