Keras
关注
分享
扫一扫
BabyKylin
这个作者很懒，什么都没留下…
展开
-
解决Keras回归模型每次预测的结果都一样？
最后一层改为outputs=Dense(1)(seq)，不要加激活函数和其他参数。原创 2021-10-02 21:27:27 · 2403 阅读 · 0 评论 -
Keras或者Tensorflow出现：Optimization loop failed: Cancelled: Operation was cancelled
方案1：文件开头加入如下代码：importtensorflowastfdevices=tf.config.experimental.list_physical_devices('GPU')tf.config.experimental.set_memory_growth(devices[0],True)方案2：重新安装CUDA和tensorflow最新版本原创 2021-10-02 19:13:45 · 6566 阅读 · 3 评论 -
向keras自定义层中传递参数
keras官方文档很简陋，只介绍了关键用法。那么来看看如何给自定义层传递其他参数。新增一个参数名叫arg。在自定义层的__init__函数中加入变量arg，并且赋值给成员变量：调用时候可以直接传递：想在自定义层中（我的是Attention层）使用，直接以成员变量的方式使用即可：...原创 2021-03-31 18:10:15 · 423 阅读 · 1 评论