How to use new pretraining model with different da

2019-07-15 01:50发布

I want to use VGG_ILSVRC_19_layers as a pretrained model in digits but with different dataset. Do I need different label files? How can I upload this model and use it for my dataset?

for the VGG 16 layers I got

ERROR: Cannot copy param 0 weights from layer 'fc6'; shape mismatch. Source param shape is 1 1 4096 25088 (102760448); target param shape is 4096 32768 (134217728). To learn this layer's parameters from scratch rather than copying from a saved net, rename the layer.

how can modify layers?

1条回答
不美不萌又怎样
2楼-- · 2019-07-15 02:27

Your labels are associated with your dataset in DIGITS - not your model. When you upload a pretrained VGG model, you'll probably need to rename your last inner product layer (see this answer) so that your model will work on N classes instead of 1000.

查看更多
登录 后发表回答