Как распечатать веса и смещения во время тренировки после каждой эпохи

Как распечатать веса и смещение в нейронной сети после каждой эпохи во время обучения

Я пробовал следующий код:

print_weights = LambdaCallback(on_epoch_end=lambda epoch, logs: print(merged.layers[1].get_weights()[0][0:5], merged.layers[2].get_weights()[0][0:5]))


merged.compile(optimizer='adam',loss=[loss1, loss2],loss_weights=[alpham, 1 - alpham], metrics=['accuracy'],callbacks = [print_weights])  

Но я получаю следующую ошибку:

ValueError: ('Некоторые ключи в session_kwargs в настоящее время не поддерживаются:% s', dict_keys (['callbacks']))