multi-network

aramis-lab · Sep 30, 2024 · 4ef55b8 · 4ef55b8
1 parent 745fab5
commit 4ef55b8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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: