小编Luc*_*uca的帖子

更新TensorFlow中的权重子集

有谁知道如何更新前向传播中使用的权重的子集(即只有一些索引)?

我的猜测是,我可以在应用compute_gradients后执行此操作,如下所示:

optimizer = tf.train.GradientDescentOptimizer(learning_rate=learning_rate)
grads_vars = optimizer.compute_gradients(loss, var_list=[weights, bias_h, bias_v])
Run Code Online (Sandbox Code Playgroud)

...然后使用元组列表执行某些操作grads_vars.

python machine-learning tensorflow

5
推荐指数
1
解决办法
5347
查看次数

标签 统计

machine-learning ×1

python ×1

tensorflow ×1