如何使用keras的load_model函数加载带参数的自定义模型？

本文共计412个文字，预计阅读时间需要2分钟。

网上的教程多数是介绍如何加载自定义模型和函数。如图所示，这个SelfAttention层是在训练过程中自定义的一个class，但如果要加载这个自定义层，需要在load_model时添加custom_objects字典。

网上的教程大多数是教大家如何加载自定义模型和函数，如下图

这个SelfAttention层是在训练过程自己定义的一个class，但如果要加载这个自定义层，需要在load_model里添加custom_objects字典，这个自定义的类，不要用import ，最好是直接复制进再训练的模型中，这些是基本教程。