AttributeError: module ‘tensorflow.python.ops.nn‘ has no attribute ‘leaky_relu‘

刀么克瑟拉莫

于 2021-04-21 14:09:30 发布

阅读量278

点赞数 1

CC 4.0 BY-SA版权

分类专栏： deeplearning python

本文链接：https://blog.youkuaiyun.com/random_repick/article/details/115950819

deeplearning 同时被 2 个专栏收录

77 篇文章

订阅专栏

python

34 篇文章

订阅专栏

本文介绍如何解决从Darknet框架迁移至TensorFlow框架过程中遇到的问题，具体为自定义LeakyReLU激活函数的实现方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

darknet转tensorflow模型时，遇到这个问题
既然tensorflow没有这个函数，我们就自己实现，把调用的地方改成调用自己的函数

import tensorflow as tf
def LeakyRelu(x, leak=0.2, name="LeakyRelu"):
     with tf.variable_scope(name):
         f1 = 0.5 * (1 + leak)
         f2 = 0.5 * (1 - leak)
         return f1 * x + f2 * tf.abs(x)