摘要:
本文介绍了如何将 PyTorch 中实现的 AlexNet 预训练模型加载到 tf.keras 中。tf.keras没有预训练好的AlexNet模型,如果我们想要在keras上用AlexNet来做迁移学习,要重新在ImageNet上跑一遍AlexNet代码?这是不现实的。其它深度学习框架,如 PyTorch,有在ImageNet上训练好的 AlexNet,我们将其它框架训练好的模型参数加载到keras中。但不同框架之间保存模型的文件有差异,需要进行一些操作。本文利用微软的开源工具 MMdnn,实现 AlexNet 模型的迁移。 阅读全文