tensorflow - 我可以逐层训练 Keras/TF 模型吗？

Question

我正在寻找训练一个大型人脸识别网络。Resnet 或 VGG-16/19。TensorFlow 1.14

我的问题是——如果我的 GPU 内存用完了——逐层训练层集是否有效？

例如，将 2 个 cnn 和 maxpooling 层训练为一组，然后以某种方式“冻结权重”并训练下一组等。

我知道我可以在 tensorflow 中训练多 GPU，但是如果我只想坚持使用一个 GPU 怎么办..

标签： tensorflowkerasgpuresnetvgg-net

Answer 1

通常的方法是使用迁移学习：使用预训练模型并针对任务对其进行微调。

对于计算机视觉的微调，一种已知的方法是仅重新训练最后几层。参见例如：
https ://www.learnopencv.com/keras-tutorial-fine-tuning-using-pre-trained-models/