From 27c582a48c46d523d903a1e59812b6b45ff4f512 Mon Sep 17 00:00:00 2001
From: Louis Bethune <l_bethune@apple.com>
Date: Fri, 21 Jun 2024 16:58:57 +0200
Subject: [PATCH] Force ICNN to adopt default initialization of its own layers

---
 src/ott/neural/networks/icnn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/ott/neural/networks/icnn.py b/src/ott/neural/networks/icnn.py
index 1adde02ce..b6f3fd049 100644
--- a/src/ott/neural/networks/icnn.py
+++ b/src/ott/neural/networks/icnn.py
@@ -23,7 +23,7 @@
 
 __all__ = ["ICNN"]
 
-DEFAULT_KERNEL_INIT = lambda *a, **k: nn.initializers.normal()(*a, **k)
+DEFAULT_KERNEL_INIT = posdef.DEFAULT_KERNEL_INIT
 DEFAULT_RECTIFIER = nn.activation.relu
 DEFAULT_ACTIVATION = nn.activation.relu