我使用以下代码行使用张量板来可视化 ANN 模型的梯度
tensorboard_callback = tf.compat.v1.keras.callbacks.TensorBoard(log_dir='./Graph', histogram_freq=1, write_graph = True, write_grads =True, write_images = False)
tensorboard_callback .set_model(model)
%tensorboard --logdir ./Graph
Run Code Online (Sandbox Code Playgroud)
我收到一条警告消息,内容为“警告:tensorflow:回调write_grads将在 TensorFlow 2.0 中被忽略TensorBoard”。
我得到张量板输出,但没有梯度。
可能的原因是什么?
(注:我使用的是2.3.0tensorflow版本)
谢谢。