如何使用pytorch仅使用vgg16的最后几层

时间:2019-05-20 19:11:40

标签: deep-learning pytorch google-colaboratory

训练VGG16的整个网络非常漫长,并且需要很多时间。 我不知道如何仅选择vgg 16的最后几层,因为我的网络仅花了6个小时以上的2个时间段,我只想在最后几层上对vgg16模型进行预训练。这可能吗 ?如果是这样,那么如何?

0 个答案:

没有答案