我需要编写一个代码,通过给定的步骤计算lambda来逐渐衰减损失函数的权重,但我不知道。任何帮助都将不胜感激。你知道吗
这是我的损失函数:
loss_A = criterion(recov_A, real_A)
loss_Final = lambda_A * loss_A + #lambda_A is a fixed number: 10
我不想把兰姆达修好。我需要在通过指定的步数后逐渐衰减lambda
# write function that computes lambda given the steps
cur_lambda = compute_lambda(step, decay_params, initial_lamdba)
Loss_Final = cur_lambda * loss_A
要衰减固定的数字取决于步数,甚至是纪元数,您可以使用以下代码,也可以将代码编写为函数,并随时调用它。你知道吗
显而易见的解决方案是:
输出:
相关问题 更多 >
编程相关推荐