我想为每个替代epoch或每5个epoch更改损失函数。我尝试使用本文中建议的损失 Package 方法。这不起作用。它永远保持current_epoch值初始化,并且更新的值不会进入损失 Package 函数,尽管该变量在on_epoch_end回调中为每个epoch结束更新。另外我尝试在on_epoch_end回调中使用model.add_loss方法,也是不行的,它只接受model.compile中初始化的损失函数,不接受model.add_loss中传入的损失函数。
qyswt5oh1#
只需多次编译模型:
for epoch in range(num_epochs): if epoch % 2 == 0: current_loss_fn = loss_fn_1 else: current_loss_fn = loss_fn_2 model.compile(optimizer=current_optimizer, loss=current_loss_fn) train...
1条答案
按热度按时间qyswt5oh1#
只需多次编译模型: