gpt4 book ai didi

python - 我可以编写具有不可训练/固定权重的 Keras Lambda 层吗?

转载 作者:行者123 更新时间:2023-12-01 07:43:17 25 4
gpt4 key购买 nike

https://keras.io/layers/writing-your-own-keras-layers/

根据此处,无状态客户操作可以包装到 Lambda 层。但如果有可训练的变量,我们需要实现我们自己的层。我想问是否可以使用 Lambda 层来实现固定权重(即预训练权重)?

如果是,如何确保权重正确加载而不是随机初始化?因为 get_weights() 不会为您提供 Lambda 层中的权重。

提前非常感谢。

最佳答案

不,Lambda 层没有来自 Keras 端的任何权重,您可以在 lambda 函数中放置一些权重,但这些权重无法使用 Keras 加载,并且对 Keras 完全不可见。

因此,如果您想使用预训练的权重,则必须实现自己的层,因此 Keras 知道这些权重,并且可以使用 load_weights 加载。

关于python - 我可以编写具有不可训练/固定权重的 Keras Lambda 层吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56581569/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com