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