From 2a3e6841853870503484c283c2bf6058e8b58f5f Mon Sep 17 00:00:00 2001 From: Albert Zeyer Date: Tue, 7 Jan 2025 01:10:49 +0100 Subject: [PATCH] small fix --- .../exp2024_04_23_baselines/model_ext/ctc_sep_net.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/users/zeyer/experiments/exp2024_04_23_baselines/model_ext/ctc_sep_net.py b/users/zeyer/experiments/exp2024_04_23_baselines/model_ext/ctc_sep_net.py index 47f44ba05..0a9b0a936 100644 --- a/users/zeyer/experiments/exp2024_04_23_baselines/model_ext/ctc_sep_net.py +++ b/users/zeyer/experiments/exp2024_04_23_baselines/model_ext/ctc_sep_net.py @@ -793,7 +793,7 @@ def backward(ctx, grad_log_probs_main, grad_log_probs_sep): ny_sep_interpolated_scaled = ny_sep_scaled * (1 - beta) + ny_main_scaled * (beta * scale_ratio) else: ny_sep_interpolated_scaled = ny_sep_scaled - return ny_main_interpolated_scaled, ny_sep_interpolated_scaled + return ny_main_interpolated_scaled, ny_sep_interpolated_scaled, None, None log_probs_main, log_probs_sep = _InterpolateGradFunc.apply(log_probs_main, log_probs_sep, alpha, beta) return log_probs_main, log_probs_sep