首页 > 解决方案 > 如何使用pytorch仅使用vgg16的最后几层

问题描述

训练VGG16的整个网络非常耗时,需要很多时间。我不知道如何只选择 vgg 16 的最后几层,因为我的网络仅在 2 个 epoch 中花费了 6 多个小时,我只想在最后几层上预训练我的 vgg16 模型。这可能吗 ?如果是这样,那怎么办?

标签: deep-learningpytorchgoogle-colaboratory

解决方案


推荐阅读