fix

MurrellGroup · Feb 10, 2024 · 869ebdb · 869ebdb
1 parent 65f1941
commit 869ebdb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))