小编Muh*_*sim的帖子

启用急切执行时不支持 tf.gradients。使用 tf.GradientTape 代替

from tensorflow.keras.applications import VGG16
from tensorflow.keras import backend as K

model = VGG16(weights='imagenet',
              include_top=False)

layer_name = 'block3_conv1'
filter_index = 0

layer_output = model.get_layer(layer_name).output
loss = K.mean(layer_output[:, :, :, filter_index])

grads = K.gradients(loss, model.input)[0]
Run Code Online (Sandbox Code Playgroud)

我无法执行grads = K.gradients(loss, model.input)[0],它产生一个错误:tf.gradients is not supported when eager execution is enabled. Use tf.GradientTape instead

python keras tensorflow

2
推荐指数
1
解决办法
4802
查看次数

标签 统计

keras ×1

python ×1

tensorflow ×1