参考:https://blog.youkuaiyun.com/wm6274/article/details/73335858
今天跑DeepFake时居然发现自己的tensorflow版本太高了,出现缺少"leaky_relu"模块。这一问题是由于版本问题出现以下报错。
正巧我在https://blog.youkuaiyun.com/wm6274/article/details/73335858这个人的文章上看到他从github上转载了 LeakyRelu函数的实现方法,那么既然缺少该函数,我们就给它补上。
第一步:
点击报错信息,进入缺少改函数的文件
LeakyRelu
添加
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)
然后修改x = tf.nn.leaky_relu(x, alpha)修改为x = LeakyRelu(x, alpha)