콜백을 사용하지 않고 손실에 액세스

Question

train_on_batch를 사용하고 수동으로 각 배치에서 호출하기 때문에 콜백을 통해 손실에 액세스 할 수 없습니다 (내 용도로 필요함).

self.model.layers[2].get_weights()[0]으로 레이어의 가중치에 액세스하는 것과 같은 각 배치 후에 손실에 액세스 할 수있는 방법이 있습니까?

콜백 정확히 train_on_batch 반환 현재 계산 된 배치의 스칼라 손실

Answer 1

기능을 가지고있다.

그래서 특별히 당신이 할 것

loss = model.train_on_batch(batch_x, batch_y)