From 4ef55b86153c88647a0e96ca790ff37732514497 Mon Sep 17 00:00:00 2001
From: camillebrianceau <camille.brianceau@gmail.com>
Date: Mon, 30 Sep 2024 17:41:26 +0200
Subject: [PATCH] multi-network

---
 clinicadl/trainer/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/clinicadl/trainer/trainer.py b/clinicadl/trainer/trainer.py
index 9231d4228..5986a9bbe 100644
--- a/clinicadl/trainer/trainer.py
+++ b/clinicadl/trainer/trainer.py
@@ -460,6 +460,7 @@ def _train_single(
             split,
             resume=resume,
             callbacks=[CodeCarbonTracker],
+            network=network,
         )
 
         if network is not None: