strategy fix

daisybio · Dec 13, 2024 · 7121036 · 7121036
1 parent 56dc761
commit 7121036
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/drevalpy/models/SimpleNeuralNetwork/utils.py b/drevalpy/models/SimpleNeuralNetwork/utils.py
@@ -253,6 +253,7 @@ def fit(
                 progress_bar,
             ],
             default_root_dir=os.path.join(os.getcwd(), "nn_baseline_checkpoints/lightning_logs/" + name),
+            strategy="ddp_find_unused_parameters_true",
             **trainer_params_copy,
         )
         if val_loader is None: