From 869ebdb978e0983fd98a637e44c3aab4a08d1c06 Mon Sep 17 00:00:00 2001
From: Lukas Billera <125421968+billera@users.noreply.github.com>
Date: Sat, 10 Feb 2024 07:02:08 +0100
Subject: [PATCH] fix

---
 src/layers.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/layers.jl b/src/layers.jl
index bffb346..579b9d4 100644
--- a/src/layers.jl
+++ b/src/layers.jl
@@ -128,7 +128,7 @@ function (ipa::Union{IPCrossA, IPA})(TiL::Tuple{AbstractArray,AbstractArray}, si
     N_frames_L = size(siL,2)
     N_frames_R = size(siR,2)
 
-    gamma_h = softplus(l.gamma_h)
+    gamma_h = min.(softplus(l.gamma_h),1f2)
 
     w_C = Typ(sqrt(2/(9*N_query_points)))
     dim_scale = Typ(1/sqrt(c))