标签: keras
我无法通过回调访问损失,因为我正在使用train_on_batch并在每个批次上手动调用它(这对我来说是必要的)。
我是否可以通过self.model.layers[2].get_weights()[0]访问每个批次后的损失,以便访问图层的权重?
self.model.layers[2].get_weights()[0]
回调有
答案 0 :(得分:0)
函数train_on_batch完全返回当前计算批次的标量损失。
特别是你会这样做
loss = model.train_on_batch(batch_x, batch_y)