分类：权重导数

2020
10-07

keras打印loss对权重的导数方式

Notes怀疑模型梯度爆炸，想打印模型loss对各权重的导数看看。如果如果fit来训练的话，可以用keras.callbacks.TensorBoard实现。但此次使用train_on_batch来训练的，用K.gradients和K.function实现。Codes以一份VAE代码为例#-*-coding:utf8-*-importkerasfromkeras.modelsimportModelfromkeras.layersimportInput,Lambda,Conv2D,MaxPooling2D,Flatten,Dense,Reshapefromkeras.lossesimportbinary_crossentr... 继续阅读 >